| D | convolution_variants_test.cc | 59 const Array4D<float> filter_array(1, 1, 1, 1, {3}); in XLA_TEST_F() local
74 const Array4D<float> filter_array(1, 1, 1, 1, {2}); in XLA_TEST_F() local
90 const Array4D<float> filter_array(1, 1, 1, 1, {2.3}); in XLA_TEST_F() local
106 const Array4D<float> filter_array(3, 2, 1, 1, {1, 2, 3, 4, 5, 6}); in XLA_TEST_F() local
121 const Array4D<float> filter_array(1, 1, 1, 2, {10, 1}); in XLA_TEST_F() local
136 const Array4D<float> filter_array(1, 1, 1, 2, {10, 1}); in XLA_TEST_F() local
151 const Array4D<float> filter_array(1, 1, 1, 2, {10, 1}); in XLA_TEST_F() local
166 const Array4D<float> filter_array(1, 1, 2, 1, {10, 1}); in XLA_TEST_F() local
181 const Array4D<float> filter_array(1, 1, 2, 2, {1000, 100, 10, 1}); in XLA_TEST_F() local
198 const Array4D<float> filter_array( in XLA_TEST_F() local
[all …]
|