Searched defs:should_decompose (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_decompose_mem_bound_mm.py | 87 def test_decompose_bmm(self, b, m, n, k, should_decompose): argument
125 def test_decompose_linear(self, m, n, k, has_bias, should_decompose): argument
170 self, m, n, k, has_bias, should_decompose argument
216 def test_decompose_mm(self, m, n, k, has_bias, should_decompose): argument
255 def test_decompose_mm_mixed_precision(self, m, n, k, has_bias, should_decompose): argument
292 def test_dynamic_shape(self, m, n, k, has_bias, should_decompose): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | all_gather_decomposer.h | 30 std::function<bool(const HloAllGatherInstruction&)> should_decompose) in AllGatherDecomposer()
|