Method _get_tensor

backends/xnnpack/operators/quant_params.py:265–268

(node)

            ), "ExportedProgram must be provided to extract per channel params"

            def _get_tensor(node):
                param = get_param_tensor(ep, node)
                assert param is not None, f"Expected to find param tensor for {node}"
                return cast(torch.Tensor, param)

            scale = _get_tensor(scale)
            zp = _get_tensor(zp)