Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_simple_test.cc483 std::array<int64, 2> input_bounds2; member
580 Array2D<float> r2_implicit_array2(spec.input_bounds2[0], in XLA_TEST_P()
581 spec.input_bounds2[1]); in XLA_TEST_P()
590 MakeR2Data(spec.input_bounds2, spec.minor2major_layout, in XLA_TEST_P()
609 float v3 = r2_implicit_array2(i % spec.input_bounds2[0], in XLA_TEST_P()
610 j % spec.input_bounds2[1]); in XLA_TEST_P()