Searched refs:reshape0 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 47 auto reshape0 = in TEST_F() local
52 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
88 auto reshape0 = variable
95 root_shape, HloOpcode::kAdd, reshape0, const1));
116 auto reshape0 = in TEST_F() local
121 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
143 auto reshape0 = in TEST_F() local
148 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
234 auto reshape0 = variable
239 root_shape, HloOpcode::kAdd, reshape0, param1));
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 116 HloInstruction* reshape0 = in TEST_F() local
120 MakeDot(ShapeUtil::MakeShape(F32, {1024, 1}), reshape0, arg1)); in TEST_F()
|