HomeSort by relevance Sort by last modified time
    Searched refs:MapWithIndexArray4D (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc 128 TEST_F(ReferenceUtilTest, MapWithIndexArray4D) {
136 auto result = ReferenceUtil::MapWithIndexArray4D(*input, subtract_index);
reference_util.h 435 return MapWithIndexArray4D(input,
446 static std::unique_ptr<Array4D<float>> MapWithIndexArray4D(
470 return MapWithIndexArray4D(
481 static std::unique_ptr<Array4D<float>> MapWithIndexArray4D(

Completed in 109 milliseconds