Searched refs:r2_array (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | broadcast_simple_test.cc | 72 Array2D<float>* r2_array, float start, float end, int seed) { in MakeR2Data() argument
74 r2_array->FillRandom(start, end, seed); in MakeR2Data()
75 auto r2_data = LiteralUtil::CreateR2FromArray2D(*r2_array).Relayout( in MakeR2Data()
575 Array2D<float> r2_array(spec.output_bounds[0], spec.output_bounds[1]); in XLA_TEST_P() local
583 &r2_array, 1.0, 2.5, 56789); in XLA_TEST_P()
606 float v2 = r2_array(i, j); in XLA_TEST_P()
|