Home
last modified time | relevance | path

Searched refs:CreateDefaultConvDimensionNumbers (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_variants_test.cc642 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 …]
H A Dconvolution_test_1d.cc244 /*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()
H A Dconvolution_dimension_numbers_test.cc103 XlaBuilder::CreateDefaultConvDimensionNumbers(); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtranspose_folding_test.cc237 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()
H A Dhlo_cost_analysis_test.cc377 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in TEST_F()
400 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in TEST_F()
1065 XlaBuilder::CreateDefaultConvDimensionNumbers(2)); in TEST_F()
H A Ddynamic_padder_test.cc346 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
389 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
H A Dhlo_evaluator_test.cc1094 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()
H A Ddynamic_dimension_inference_test.cc459 auto dnums = XlaBuilder::CreateDefaultConvDimensionNumbers(0); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.cc52 XlaBuilder::CreateDefaultConvDimensionNumbers(1)); in ConvArray3D()
99 XlaBuilder::CreateDefaultConvDimensionNumbers()); in ConvArray4D()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc1453 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
H A Dxla_builder.h273 static ConvolutionDimensionNumbers CreateDefaultConvDimensionNumbers(