Home
last modified time | relevance | path

Searched refs:output_bounds (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_simple_test.cc305 std::array<int64, 3> output_bounds; member
331 Array3D<float> r3_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P()
332 spec.output_bounds[2]); in XLA_TEST_P()
337 MakeR3Data(spec.output_bounds, spec.minor2major_layout, &r3_shape, in XLA_TEST_P()
348 Array3D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P()
349 spec.output_bounds[2]); in XLA_TEST_P()
480 std::array<int64, 2> output_bounds; member
577 Array2D<float> r2_array(spec.output_bounds[0], spec.output_bounds[1]); in XLA_TEST_P()
584 MakeR2Data(spec.output_bounds, spec.minor2major_layout, &r2_shape, in XLA_TEST_P()
603 Array2D<float> expected_array(spec.output_bounds[0], spec.output_bounds[1]); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h554 int64 output_bounds[3]; in PadArray3D() local
564 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
568 Array3D<NativeT> result(output_bounds[0], output_bounds[1], in PadArray3D()
569 output_bounds[2]); in PadArray3D()
571 for (indices[0] = 0; indices[0] < output_bounds[0]; ++indices[0]) { in PadArray3D()
572 for (indices[1] = 0; indices[1] < output_bounds[1]; ++indices[1]) { in PadArray3D()
573 for (indices[2] = 0; indices[2] < output_bounds[2]; ++indices[2]) { in PadArray3D()
578 bool in_high_padding = indices[i] >= output_bounds[i] - pad_high[i]; in PadArray3D()
613 int64 output_bounds[4]; in PadArray4D() local
621 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray4D()
[all …]