HomeSort by relevance Sort by last modified time
    Searched refs:reshape2 (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
instruction_fusion_test.cc 75 HloInstruction* reshape2 = builder.AddInstruction( local
80 EXPECT_EQ(reshape2, computation->root_instruction());
112 auto reshape2 = builder.AddInstruction(HloInstruction::CreateReshape( local
117 EXPECT_EQ(reshape2, computation->root_instruction());
  /external/tensorflow/tensorflow/compiler/xla/service/
reshape_mover_test.cc 156 // +- reshape2 - param2
164 // Verifies that the reshape1 and reshape2 sink past select:
188 auto reshape2 = local
192 root_shape, HloOpcode::kSelect, const0, reshape1, reshape2));
197 op::Select(const0, reshape1, reshape2));
303 // reshape2
516 // +- reshape2 - param2
542 auto reshape2 = local
547 reshape2, reshape3));

Completed in 57 milliseconds