Home
last modified time | relevance | path

Searched refs:_qconv2d_unary_cpu_test_helper (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_mkldnn_pattern_matcher.py715 def _qconv2d_unary_cpu_test_helper( member in TestPatternMatcher
767 self._qconv2d_unary_cpu_test_helper()
777 self._qconv2d_unary_cpu_test_helper(int8_mixed_bf16=True)
786 self._qconv2d_unary_cpu_test_helper(unary_op=torch.nn.ReLU6())
795 self._qconv2d_unary_cpu_test_helper(unary_op=torch.nn.Hardtanh())
808 self._qconv2d_unary_cpu_test_helper(
821 self._qconv2d_unary_cpu_test_helper(unary_op=torch.nn.Hardswish())
835 self._qconv2d_unary_cpu_test_helper(
848 self._qconv2d_unary_cpu_test_helper(unary_op=torch.nn.SiLU())
862 self._qconv2d_unary_cpu_test_helper(