Searched refs:ReduceWindow4DAdd (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | reduce_window_test.cc | 170 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 2, 1}, in XLA_TEST_P() 185 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 2, 1}, in XLA_TEST_P() 199 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 1, 1}, in XLA_TEST_P() 216 auto res = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() 237 auto result = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() 274 auto result = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() 299 auto result = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() 317 auto result = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() 375 auto res = ReferenceUtil::ReduceWindow4DAdd(input_array, 0.0f, {1, 1, 7, 1}, in XLA_TEST_P() 436 auto res = ReferenceUtil::ReduceWindow4DAdd( in XLA_TEST_P() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | reference_util.h | 186 static std::unique_ptr<Array4D<float>> ReduceWindow4DAdd(
|
D | reference_util.cc | 300 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::ReduceWindow4DAdd( in ReduceWindow4DAdd() function in xla::ReferenceUtil
|