| D | reduce_window_test.cc | 72 void ReduceWindowAdd(const XlaOp& input, in ReduceWindowAdd() function in xla::__anon014212b70111::ReduceWindowTest
159 ReduceWindowAdd(input, {1, 1, 2, 1}, {1, 1, 1, 1}, padding); in XLA_TEST_P()
174 ReduceWindowAdd(input, {1, 1, 2, 1}, {1, 1, 1, 1}, padding); in TEST_P()
188 ReduceWindowAdd(input, {1, 1, 1, 1}, {1, 2, 2, 1}, padding); in TEST_P()
205 ReduceWindowAdd(input, {1, 1, lrn_diameter, 1}, {1, 1, 1, 1}, padding); in TEST_P()
225 ReduceWindowAdd(input_data_handle, {win_len, win_len, 1, 1}, in TEST_P()
248 ReduceWindowAdd(input_data_handle, {win_len, win_len, 1, 1}, in TEST_P()
273 ReduceWindowAdd(input_data_handle, {win_len, win_len, 1, 1}, in TEST_P()
292 ReduceWindowAdd(input_data_handle, {1, 1, 1, 11}, {1, 1, 1, 1}, padding); in TEST_P()
349 ReduceWindowAdd(input, {1, 1, 7, 1}, {1, 4, 1, 1}, padding); in TEST_P()
[all …]
|