Home
last modified time | relevance | path

Searched refs:MapWithIndexArray4D (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.h448 return MapWithIndexArray4D( in MapArray4D()
459 static std::unique_ptr<Array4D<float>> MapWithIndexArray4D( in MapWithIndexArray4D() function
483 return MapWithIndexArray4D( in MapArray4D()
495 static std::unique_ptr<Array4D<float>> MapWithIndexArray4D( in MapWithIndexArray4D() function
H A Dreference_util_test.cc137 TEST_F(ReferenceUtilTest, MapWithIndexArray4D) { in TEST_F() argument
145 auto result = ReferenceUtil::MapWithIndexArray4D(*input, subtract_index); in TEST_F()