Searched refs:Reduce3DTo2D (Results 1 – 3 of 3) sorted by relevance
157 static std::unique_ptr<Array2D<float>> Reduce3DTo2D(
679 /* static */ std::unique_ptr<Array2D<float>> ReferenceUtil::Reduce3DTo2D( in Reduce3DTo2D() function in xla::ReferenceUtil
831 ReferenceUtil::Reduce3DTo2D(input_array, 0.0f, GetParam().reduce_dims, in XLA_TEST_P()