Searched defs:use_relu (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_mkldnn_pattern_matcher.py | 868 def _qconv2d_add_cpu_test_helper(self, use_relu=False, int8_mixed_bf16=False): argument
886 use_relu, argument
1711 def _qlinear_add_cpu_test_helper(self, use_relu=False, int8_mixed_bf16=False): argument
1744 use_relu, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 774 use_relu, argument
|
| /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
3936 use_relu, use_multi_dim_input, use_channelwise): argument
4064 use_relu, use_multi_dim_input, use_channelwise): argument
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 1244 def _get_conv_transpose_test_cases(self, use_relu, is_reference): argument
6363 def __init__(self, use_relu=False): argument
6395 def __init__(self, use_relu=False): argument
|