HomeSort by relevance Sort by last modified time
    Searched refs:input_bounds (Results 1 - 2 of 2) sorted by null

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

Completed in 1401 milliseconds