Home
last modified time | relevance | path

Searched defs:per_channel (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dfully-connected-operator-tester.h85 inline FullyConnectedOperatorTester& per_channel(bool per_channel) { in per_channel() argument
90 inline bool per_channel() const { in per_channel() function
H A Ddeconvolution-operator-tester.h131 inline DeconvolutionOperatorTester& per_channel(bool per_channel) { in per_channel() function
136 inline bool per_channel() const { in per_channel() function
H A Dconvolution-operator-tester.h160 inline ConvolutionOperatorTester& per_channel(bool per_channel) { in per_channel() function
165 inline bool per_channel() const { in per_channel() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dconvolution.c43 static enum pytorch_qnnp_status pytorch_qnnp_create_convolution_ndhwc_q8( in pytorch_qnnp_create_convolution_ndhwc_q8()
570 bool per_channel, in pytorch_qnnp_create_convolution2d_nhwc_q8()
602 enum pytorch_qnnp_status pytorch_qnnp_create_convolution3d_ndhwc_q8( in pytorch_qnnp_create_convolution3d_ndhwc_q8()
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_ondevice_quantization.py192 def _validate_quant_dequant_nodes(self, model, num_nodes, per_channel=0): argument
259 def _validate_packed_params(self, model, num_nodes, per_channel=0): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DQnnpackUtils.h89 bool per_channel() const { in per_channel() function
362 bool per_channel() const { in per_channel() function
H A DXnnpackUtils.h40 enum xnn_status xnnp_create_convolution2d_nhwc( in xnnp_create_convolution2d_nhwc()
H A Dqlinear.cpp743 static bool can_use_xnnp(c10::ScalarType dtype, bool per_channel) { in can_use_xnnp()
H A Dqconv.cpp1031 bool per_channel, in can_use_xnnp()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
H A Dconvolution.cc22 static void convolution_q8(benchmark::State& state, const char* net, bool per_channel=false) { in convolution_q8()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
H A Doperator.h129 bool per_channel; member
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_module.py100 per_channel, **post_ops_kwargs): argument
H A Dtest_quantized_tensor.py1446 def __init__(self, per_channel): argument
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_fs_reg_allocate.cpp1027 const bool per_channel = in spill_reg() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_fs_reg_allocate.cpp1085 const bool per_channel = in spill_reg() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantization_utils.cc710 TfLiteStatus QuantizeWeight(ModelT* model, TensorT* tensor, bool per_channel, in QuantizeWeight()
H A Dquantization_utils_test.cc562 bool per_channel = false; in TEST_F() local