| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion_test.cc | 80 HloInstruction* reshape2 = builder.AddInstruction( local
85 EXPECT_EQ(reshape2, computation->root_instruction());
117 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape( local
122 EXPECT_EQ(reshape2, computation->root_instruction());
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| reshape_mover_test.cc | 164 // +- reshape2 - param2
172 // Verifies that the reshape1 and reshape2 sink past select:
198 auto reshape2 = local
202 root_shape, HloOpcode::kSelect, const0, reshape1, reshape2));
207 op::Select(const0, reshape1, reshape2));
315 // reshape2
482 // +- reshape2 - param2
509 auto reshape2 = local
514 reshape2, reshape3));
561 reshape2 = f32[8,20]{1,0} reshape(param3
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_instruction_fusion_test.cc | 319 HloInstruction* reshape2 = local
322 HloInstruction::CreateUnary(result_shape, HloOpcode::kNegate, reshape2));
[all...] |
