Searched defs:input_bounds (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | reference_util.h | 541 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 D | broadcast_simple_test.cc | 229 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