Searched refs:CreateDefaultConvDimensionNumbers (Results 1 – 11 of 11) sorted by relevance
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()
284 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in RunTest()320 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in XLA_TEST_F()351 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in XLA_TEST_F()385 /*dimension_numbers=*/builder.CreateDefaultConvDimensionNumbers(1)); in RunTest()
230 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()286 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()348 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()415 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(); in TEST_F()
118 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()161 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
979 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1201 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1265 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()1337 XlaBuilder::CreateDefaultConvDimensionNumbers(2); in TEST_P()
584 XlaBuilder::CreateDefaultConvDimensionNumbers(2)); in TEST_F()
302 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
67 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in ConvArray3D()114 XlaBuilder::CreateDefaultConvDimensionNumbers()); in ConvArray4D()
1105 CreateDefaultConvDimensionNumbers(window_strides.size()), in Conv()1115 CreateDefaultConvDimensionNumbers(window_strides.size()), in ConvWithGeneralPadding()2646 XlaBuilder::CreateDefaultConvDimensionNumbers(int num_spatial_dims) { in CreateDefaultConvDimensionNumbers() function in xla::XlaBuilder
189 static ConvolutionDimensionNumbers CreateDefaultConvDimensionNumbers(