Searched defs:xpose (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bfloat16_propagation_test.cc | 117 HloInstruction* xpose = in TEST_F() local
178 HloInstruction* xpose = in TEST_F() local
280 HloInstruction* xpose = in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_test.cc | 7320 auto xpose = AllOf(op::Shape("f32[2,4,4,4,4]"), op::Transpose(all_to_all)); in TEST_F() local
7637 auto xpose = AllOf(op::Shape("f32[2,2,4,12,32]"), op::Transpose(all_to_all)); in TEST_F() local
|