Searched refs:ReduceToRowArray2D (Results 1 – 3 of 3) sorted by relevance
81 TEST_F(ReferenceUtilTest, ReduceToRowArray2D) { in TEST_F() argument83 auto result = ReferenceUtil::ReduceToRowArray2D(*matrix_, 0.0f, add); in TEST_F()
123 static std::unique_ptr<std::vector<float>> ReduceToRowArray2D(
588 ReferenceUtil::ReduceToRowArray2D( in ReduceToRowArray2D() function in xla::ReferenceUtil