Searched refs:input_bounds (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_simple_test.cc | 307 std::array<int64, 3> input_bounds; member
333 Array3D<float> r3_implicit_array(spec.input_bounds[0], spec.input_bounds[1], in XLA_TEST_P()
334 spec.input_bounds[2]); in XLA_TEST_P()
340 MakeR3Data(spec.input_bounds, spec.minor2major_layout, &r3_implicit_shape, in XLA_TEST_P()
351 float r3_implicit = r3_implicit_array(indices[0] % spec.input_bounds[0], in XLA_TEST_P()
352 indices[1] % spec.input_bounds[1], in XLA_TEST_P()
353 indices[2] % spec.input_bounds[2]); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 550 const int64 input_bounds[] = {operand.n1(), operand.n2(), operand.n3()}; in PadArray3D() local
564 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray3D()
565 (input_bounds[i] - 1) * pad_interior[i]; in PadArray3D()
608 const int64 input_bounds[] = {operand.n1(), operand.n2(), operand.n3(), in PadArray4D() local
621 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] + in PadArray4D()
622 (input_bounds[i] - 1) * pad_interior[i]; in PadArray4D()
|