Searched refs:CreateDefaultConvDimensionNumbers (Results 1 – 11 of 11) sorted by relevance
642 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()661 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()684 /*rhs_dilation=*/{}, XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()706 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()725 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()744 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()764 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()789 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()1178 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()1240 XlaBuilder::CreateDefaultConvDimensionNumbers()); in XLA_TEST_F()[all …]
244 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in RunTest()280 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in XLA_TEST_F()311 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in XLA_TEST_F()345 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in RunTest()
103 XlaBuilder::CreateDefaultConvDimensionNumbers(); in XLA_TEST_F()
237 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()295 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()359 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()428 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()
377 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in TEST_F()400 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in TEST_F()1065 XlaBuilder::CreateDefaultConvDimensionNumbers(2)); in TEST_F()
346 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()389 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
1094 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1316 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1380 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1452 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()
459 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
52 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in ConvArray3D()99 XlaBuilder::CreateDefaultConvDimensionNumbers()); in ConvArray4D()
1453 CreateDefaultConvDimensionNumbers(window_strides.size()), in Conv()1465 CreateDefaultConvDimensionNumbers(window_strides.size()), in ConvWithGeneralPadding()3836 XlaBuilder::CreateDefaultConvDimensionNumbers(int num_spatial_dims) { in CreateDefaultConvDimensionNumbers() function in xla::XlaBuilder
273 static ConvolutionDimensionNumbers CreateDefaultConvDimensionNumbers(