Searched defs:output_bounds (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ | ||
H A D | broadcast_simple_test.cc | 227 std::array<int64, 3> output_bounds; member in struct:xla::__anon25721::R3ImplicitBroadcastSpec 253 Array3D<float> r3_array(spec.output_bounds[0], spec.output_bounds[1], 254 spec.output_bounds[2]); 259 MakeR3Data(spec.output_bounds, spec.minor2major_layout, &r3_shape, 270 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1], 271 spec.output_bounds[2]); 401 std::array<int64, 2> output_bounds; member in struct:xla::__anon25721::R2ImplicitBroadcastSpec 498 Array2D<float> r2_array(spec.output_bounds[ [all...] |
Completed in 164 milliseconds