| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 49 auto reshape1 = in TEST_F() local
52 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
118 auto reshape1 = in TEST_F() local
121 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
145 auto reshape1 = in TEST_F() local
148 root_shape, HloOpcode::kAdd, reshape0, reshape1)); in TEST_F()
193 auto reshape1 = variable
202 root_shape, HloOpcode::kSelect, const0, reshape1, reshape2));
207 op::Select(const0, reshape1, reshape2));
276 auto reshape1 = variable
[all …]
|
| D | pattern_matcher_test.cc | 901 auto reshape1 = in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 1868 auto reshape1 = builder.AddInstruction( in TEST_F() local
1871 ShapeUtil::MakeShape(F32, {1, 6}), reshape1, {1})); in TEST_F()
2712 HloInstruction* reshape1 = in TEST_F() local
2717 ShapeUtil::MakeShape(F32, {1, 2, 1, 1, 2, 1}), reshape1)); in TEST_F()
2905 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
2908 ShapeUtil::MakeShape(F32, {1, 2, 3, 5, 1}), reshape1, {0, 3, 2})); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 350 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local
353 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
365 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
368 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
380 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
383 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
395 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local
398 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
410 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
413 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
[all …]
|
| D | dynamism_inference_test.cc | 192 auto reshape1 = Reshape(slice1, {}); in TEST_F() local
193 auto tuple_2 = Tuple(&b, {reshape0, reshape1}); in TEST_F()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_input_test.py | 684 reshape1 = array_ops.reshape(tile, [-1, 2])
688 reshape1,
693 reshape3 = array_ops.reshape(reshape1,
696 reshape4 = array_ops.reshape(reshape1,
698 return [reshape1, reshape2, reshape3, reshape4]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 432 HloInstruction* reshape1 = builder.AddInstruction( in TEST_F() local
435 HloInstruction::CreateUnary(result_shape, HloOpcode::kNegate, reshape1)); in TEST_F()
|