Searched refs:r3_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_simple_test.cc | 58 Shape* r3_shape, in MakeR3Data() argument
61 *r3_shape = ShapeUtil::MakeShapeWithLayout(F32, bounds, minor_to_major); in MakeR3Data()
330 Shape r3_shape, r3_implicit_shape; in XLA_TEST_P() local
337 MakeR3Data(spec.output_bounds, spec.minor2major_layout, &r3_shape, in XLA_TEST_P()
345 auto r3_parameter = Parameter(&builder, 1, r3_shape, "input"); in XLA_TEST_P()
|