Home
last modified time | relevance | path

Searched refs:ReduceToColArray2D (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc73 TEST_F(ReferenceUtilTest, ReduceToColArray2D) { in TEST_F() argument
75 auto result = ReferenceUtil::ReduceToColArray2D(*matrix_, 0.0f, add); in TEST_F()
Dreference_util.h116 static std::unique_ptr<std::vector<float>> ReduceToColArray2D(
Dreference_util.cc571 ReferenceUtil::ReduceToColArray2D( in ReduceToColArray2D() function in xla::ReferenceUtil