Searched refs:minor2major_layout (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_simple_test.cc | 306 std::array<int64, 3> minor2major_layout; member
337 MakeR3Data(spec.output_bounds, spec.minor2major_layout, &r3_shape, in XLA_TEST_P()
340 MakeR3Data(spec.input_bounds, spec.minor2major_layout, &r3_implicit_shape, in XLA_TEST_P()
481 std::array<int64, 2> minor2major_layout; member
584 MakeR2Data(spec.output_bounds, spec.minor2major_layout, &r2_shape, in XLA_TEST_P()
587 MakeR2Data(spec.input_bounds1, spec.minor2major_layout, in XLA_TEST_P()
590 MakeR2Data(spec.input_bounds2, spec.minor2major_layout, in XLA_TEST_P()
|