Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util_test.cc72 TEST_F(ReferenceUtilTest, ReduceToColArray2D) { in TEST_F() argument
74 auto result = ReferenceUtil::ReduceToColArray2D(*matrix_, 0.0f, add); in TEST_F()
H A Dreference_util.h114 static std::unique_ptr<std::vector<float>> ReduceToColArray2D(
H A Dreference_util.cc508 ReferenceUtil::ReduceToColArray2D( in ReduceToColArray2D() function in xla::ReferenceUtil