Searched refs:reshape3 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 209 auto reshape3 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() local
214 ShapeUtil::MakeShape(F32, {2, 3}), {reshape3, const4}, 0)); in XLA_TEST_F()
237 const4, reshape3, add2, const1, const0}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 511 auto reshape3 = in TEST_F() local
514 reshape2, reshape3)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 347 HloInstruction* reshape3 = builder.AddInstruction( in TEST_F() local
351 dynamic_slice_shape, reshape3, {param1, param2}, {4, 4})); in TEST_F()
|