Searched refs:partial_dot_pattern (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_test.cc | 4522 const auto partial_dot_pattern = in TEST_F() local
4529 partial_dot_pattern)), in TEST_F()
4530 partial_dot_pattern), in TEST_F()
4769 const auto partial_dot_pattern = in TEST_F() local
4776 partial_dot_pattern)), in TEST_F()
4777 partial_dot_pattern), in TEST_F()
5046 const auto partial_dot_pattern = in TEST_F() local
5052 partial_dot_pattern, op::Constant(), in TEST_F()
5056 op::DynamicUpdateSlice(intermediate_output1, partial_dot_pattern, in TEST_F()
5060 op::DynamicUpdateSlice(intermediate_output2, partial_dot_pattern, in TEST_F()
[all …]
|