Searched defs:use_channelwise (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 3126 use_bias, use_relu, use_multi_dim_input, use_channelwise, reduce_range): argument
3755 post_op, use_multi_dim_input, use_channelwise, **post_op_kwargs): argument
3936 use_relu, use_multi_dim_input, use_channelwise): argument
4064 use_relu, use_multi_dim_input, use_channelwise): argument
4144 def test_qlinear_unpack(self, W, use_channelwise): argument
4928 use_bias, use_channelwise, use_transpose, argument
5013 Y_zero_point, use_bias, post_op, use_channelwise, use_transpose, argument
5166 use_channelwise, argument
5247 use_channelwise, argument
5454 use_channelwise, argument
[all …]
|
| H A D | test_quantized_functional.py | 38 Y_scale, Y_zero_point, use_bias, use_channelwise, argument
91 use_bias, use_channelwise, qengine, argument
143 use_bias, use_channelwise, qengine, argument
200 use_channelwise, qengine, argument
|
| H A D | test_quantized_module.py | 270 use_bias, post_op, use_channelwise, X2_scale=1.0, X2_zero_point=0): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 241 W_zero_point, use_bias, use_channelwise, argument
|