Function ArrayForDocR3Tests

tensorflow/compiler/xla/tests/reshape_test.cc:383–388

The following tests use the same input 3D array; they test the examples we show for the Reshape operation in the operation_semantics document. TODO(b/34503277): find a way to show this code in the documentation without duplication on the TF documentation server.

// TODO(b/34503277): find a way to show this code in the documentation without
// duplication on the TF documentation server.
static Array3D<float> ArrayForDocR3Tests() {
  return Array3D<float>({{{10, 11, 12}, {15, 16, 17}},
                         {{20, 21, 22}, {25, 26, 27}},
                         {{30, 31, 32}, {35, 36, 37}},
                         {{40, 41, 42}, {45, 46, 47}}});
}

XLA_TEST_P(ReshapeTest, DocR3_R1_Collapse_012) {
  XlaBuilder builder(TestName());