| D | convolution_variants_test.cc | 60 const Array4D<float> filter_array(1, 1, 1, 1, {3}); in XLA_TEST_F() local
75 const Array4D<float> filter_array(1, 1, 1, 1, {2}); in XLA_TEST_F() local
91 const Array4D<float> filter_array(1, 1, 1, 1, {2.3}); in XLA_TEST_F() local
107 const Array4D<float> filter_array(3, 2, 1, 1, {1, 2, 3, 4, 5, 6}); in XLA_TEST_F() local
122 const Array4D<float> filter_array(1, 1, 1, 2, {10, 1}); in XLA_TEST_F() local
137 const Array4D<float> filter_array(1, 1, 1, 2, {10, 1}); in XLA_TEST_F() local
152 const Array4D<float> filter_array(1, 1, 1, 2, {10, 1}); in XLA_TEST_F() local
167 const Array4D<float> filter_array(1, 1, 2, 1, {10, 1}); in XLA_TEST_F() local
182 const Array4D<float> filter_array(1, 1, 2, 2, {1000, 100, 10, 1}); in XLA_TEST_F() local
199 const Array4D<float> filter_array( in XLA_TEST_F() local
[all …]
|