| D | conv_canonicalization_test.cc | 52 static constexpr int kBatchSize = 50; member in xla::cpu::ConvCanonicalizationTest
64 kInputFeatureCount, kBatchSize, kInputSize, kInputSize)))); in TEST_F()
86 F32, {kOutputFeatureCount, kBatchSize, output_size, output_size}), in TEST_F()
127 kBatchSize, kInputSize, kInputSize, kInputFeatureCount)))); in TEST_F()
149 F32, {kBatchSize, output_size, output_size, kOutputFeatureCount}), in TEST_F()
|