Searched refs:ReduceWindow4DAdd (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | reduce_window_test.cc | 161 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 2, 1}, in XLA_TEST_P() 176 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 2, 1}, in TEST_P() 190 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 1, 1}, in TEST_P() 207 auto res = ReferenceUtil::ReduceWindow4DAdd( in TEST_P() 228 auto result = ReferenceUtil::ReduceWindow4DAdd( in TEST_P() 251 auto result = ReferenceUtil::ReduceWindow4DAdd( in TEST_P() 276 auto result = ReferenceUtil::ReduceWindow4DAdd( in TEST_P() 294 auto result = ReferenceUtil::ReduceWindow4DAdd( in TEST_P() 351 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 7, 1}, in TEST_P() 411 auto res = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util.h | 189 static std::unique_ptr<Array4D<float>> ReduceWindow4DAdd(
|
D | reference_util.cc | 366 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::ReduceWindow4DAdd( in ReduceWindow4DAdd() function in xla::ReferenceUtil
|