Searched refs:input_subtuple (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 | 5645 auto input_subtuple = in TEST_F() local
5650 lhs, rhs, input_subtuple, op::Broadcast(), op::Constant())))), in TEST_F()
5733 auto input_subtuple = in TEST_F() local
5738 lhs, rhs, input_subtuple, op::Broadcast(), op::Constant())))), in TEST_F()
5837 auto input_subtuple = in TEST_F() local
5841 op::Tuple(lhs, rhs, input_subtuple, in TEST_F()
5980 auto input_subtuple = in TEST_F() local
5985 op::While(op::Tuple(lhs, rhs, input_subtuple, op::Broadcast(), in TEST_F()
6080 auto input_subtuple = in TEST_F() local
6085 lhs, rhs, input_subtuple, op::CollectivePermute(rhs), in TEST_F()
[all …]
|