Searched refs:MapWithIndexArray2D (Results 1 – 4 of 4) sorted by relevance
40 auto aexpected = ReferenceUtil::MapWithIndexArray2D( in TEST_F()56 auto aexpected = ReferenceUtil::MapWithIndexArray2D( in TEST_F()72 auto aexpected = ReferenceUtil::MapWithIndexArray2D( in TEST_F()88 auto aexpected = ReferenceUtil::MapWithIndexArray2D( in TEST_F()
113 TEST_F(ReferenceUtilTest, MapWithIndexArray2D) { in TEST_F() argument117 auto result = ReferenceUtil::MapWithIndexArray2D(*matrix_, add_index); in TEST_F()
439 static std::unique_ptr<Array2D<float>> MapWithIndexArray2D(
704 /* static */ std::unique_ptr<Array2D<float>> ReferenceUtil::MapWithIndexArray2D( in MapWithIndexArray2D() function in xla::ReferenceUtil