Searched defs:MapWithIndexArray2D (Results 1 – 2 of 2) sorted by relevance
113 TEST_F(ReferenceUtilTest, MapWithIndexArray2D) { in TEST_F() argument
704 /* static */ std::unique_ptr<Array2D<float>> ReferenceUtil::MapWithIndexArray2D( in MapWithIndexArray2D() function in xla::ReferenceUtil