Searched refs:r3_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 | 59 Array3D<float>* r3_array, float start, float end, int seed) { in MakeR3Data() argument
61 r3_array->FillRandom(start, end, seed); in MakeR3Data()
62 auto r3_data = LiteralUtil::CreateR3FromArray3D(*r3_array).Relayout( in MakeR3Data()
329 Array3D<float> r3_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P() local
336 &r3_array, 1.0, 2.5, 56789); in XLA_TEST_P()
352 float r3 = r3_array(indices[0], indices[1], indices[2]); in XLA_TEST_P()
|