Searched refs:reshape1 (Results 1 – 6 of 6) sorted by relevance
| /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 | instruction_fusion_test.cc | 117 auto reshape1 = builder.AddInstruction( in TEST_F() local
122 EXPECT_EQ(reshape1, computation->root_instruction()); in TEST_F()
133 auto reshape1 = builder.AddInstruction( in TEST_F() local
138 EXPECT_EQ(reshape1, computation->root_instruction()); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 1504 auto reshape1 = builder.AddInstruction( in TEST_F() local
1507 ShapeUtil::MakeShape(F32, {1, 6}), reshape1, {1})); in TEST_F()
2167 HloInstruction* reshape1 = in TEST_F() local
2172 ShapeUtil::MakeShape(F32, {1, 2, 1, 1, 2, 1}), reshape1)); in TEST_F()
2270 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in TEST_F() local
2273 ShapeUtil::MakeShape(F32, {1, 2, 3, 5, 1}), reshape1, {0, 3, 2})); in TEST_F()
|
| D | pattern_matcher_test.cc | 905 auto reshape1 = in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 351 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local
354 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
366 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
369 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
381 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
384 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
396 auto reshape1 = builder.AddInstruction(HloInstruction::CreateReshape( in XLA_TEST_F() local
399 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
411 auto reshape1 = builder.AddInstruction( in XLA_TEST_F() local
414 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 430 HloInstruction* reshape1 = builder.AddInstruction( in TEST_F() local
433 HloInstruction::CreateUnary(result_shape, HloOpcode::kNegate, reshape1)); in TEST_F()
|