Searched defs:xpose (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 116 HloInstruction* xpose = in TEST_F() local
177 HloInstruction* xpose = in TEST_F() local
279 HloInstruction* xpose = in TEST_F() local
570 HloInstruction* xpose = in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 5316 auto xpose = AllOf(op::Shape("f32[2,4,4,4,4]"), op::Transpose(all_to_all)); in TEST_F() local
5634 auto xpose = AllOf(op::Shape("f32[2,2,4,12,32]"), op::Transpose(all_to_all)); in TEST_F() local
|