Searched refs:input_reshape (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dconv_canonicalization_test.cc99 const HloInstruction* input_reshape = canonical_conv->operand(0); local
100 EXPECT_EQ(HloOpcode::kTranspose, input_reshape->opcode());
104 // The input is in CNHW order. input_reshape should produce
106 EXPECT_THAT(input_reshape->dimensions(), ElementsAre(1, 2, 3, 0));
111 // input_reshape's order). output_reshape should restore that order to the

Completed in 266 milliseconds