Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc128 TEST_F(ReferenceUtilTest, MapWithIndexArray4D) { in TEST_F() argument
136 auto result = ReferenceUtil::MapWithIndexArray4D(*input, subtract_index); in TEST_F()
Dreference_util.h436 return MapWithIndexArray4D(input, in MapArray4D()
447 static std::unique_ptr<Array4D<float>> MapWithIndexArray4D( in MapWithIndexArray4D() function
471 return MapWithIndexArray4D( in MapArray4D()
482 static std::unique_ptr<Array4D<float>> MapWithIndexArray4D( in MapWithIndexArray4D() function