Searched refs:r3_array (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | broadcast_simple_test.cc | 59 Array3D<float>* r3_array, float start, in MakeR3Data() argument
62 r3_array->FillRandom(start, end, seed); in MakeR3Data()
63 auto r3_data = LiteralUtil::CreateR3FromArray3D(*r3_array).Relayout( in MakeR3Data()
331 Array3D<float> r3_array(spec.output_bounds[0], spec.output_bounds[1], in XLA_TEST_P() local
338 &r3_array, 1.0, 2.5, 56789); in XLA_TEST_P()
354 float r3 = r3_array(indices[0], indices[1], indices[2]); in XLA_TEST_P()
|