Searched defs:input_bounds (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.h541 const std::vector<int64> input_bounds = {operand.n1(), operand.n2(), local
556 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] +
557 (input_bounds[i] - 1) * pad_interior[i];
600 const std::vector<int64> input_bounds = {operand.n1(), operand.n2(), local
613 output_bounds[i] = pad_low[i] + input_bounds[i] + pad_high[i] +
614 (input_bounds[i] - 1) * pad_interior[i];
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbroadcast_simple_test.cc229 std::array<int64, 3> input_bounds; member in struct:xla::__anon25721::R3ImplicitBroadcastSpec
255 Array3D<float> r3_implicit_array(spec.input_bounds[0], spec.input_bounds[1],
256 spec.input_bounds[2]);
262 MakeR3Data(spec.input_bounds, spec.minor2major_layout, &r3_implicit_shape,
273 float r3_implicit = r3_implicit_array(indices[0] % spec.input_bounds[0],
274 indices[1] % spec.input_bounds[1],
275 indices[2] % spec.input_bounds[2]);

Completed in 575 milliseconds