Searched refs:CreateDefaultConvDimensionNumbers (Results 1 – 11 of 11) sorted by relevance
248 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in RunTest()284 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in XLA_TEST_F()315 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in XLA_TEST_F()349 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in RunTest()
643 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()662 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()685 /*rhs_dilation=*/{}, XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()707 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()726 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()745 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()765 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()790 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()1179 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()1241 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()[all …]
106 XlaBuilder::CreateDefaultConvDimensionNumbers(); in XLA_TEST_F()
231 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()288 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()351 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()419 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()
333 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in TEST_F()356 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in TEST_F()895 XlaBuilder::CreateDefaultConvDimensionNumbers(2)); in TEST_F()
274 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()317 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
1056 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1278 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1342 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1414 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()
452 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
52 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in ConvArray3D()99 XlaBuilder::CreateDefaultConvDimensionNumbers()); in ConvArray4D()
1380 CreateDefaultConvDimensionNumbers(window_strides.size()), in Conv()1392 CreateDefaultConvDimensionNumbers(window_strides.size()), in ConvWithGeneralPadding()3888 XlaBuilder::CreateDefaultConvDimensionNumbers(int num_spatial_dims) { in CreateDefaultConvDimensionNumbers() function in xla::XlaBuilder
253 static ConvolutionDimensionNumbers CreateDefaultConvDimensionNumbers(