Home
last modified time | relevance | path

Searched refs:input_spatial_dimensions (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc47 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
48 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
53 dnums.input_spatial_dimensions().begin(), in StreamExecutorConvLayoutsToXlaLayouts()
54 dnums.input_spatial_dimensions().end()); in StreamExecutorConvLayoutsToXlaLayouts()
Dcudnn_conv_padding_legalization.cc64 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
65 int64 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
98 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) { in MaybePaddedAndSlicedInput()
99 int64 dim = conv_dnums.input_spatial_dimensions(i); in MaybePaddedAndSlicedInput()
236 int64 dim = backward_conv_dnums.input_spatial_dimensions(i); in CanonicalizeBackwardFilterConvolution()
Dcudnn_conv_rewriter.cc120 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter()
327 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput()
Dcudnn_conv_runner.cc190 input_shape.dimensions(dnums.input_spatial_dimensions(dim))); in RunCudnnConvImpl()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc389 dnums.input_spatial_dimensions(0), in TEST_F()
390 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
392 dnums.input_spatial_dimensions(1), in TEST_F()
393 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
456 dnums.input_spatial_dimensions(0), in TEST_F()
457 new_conv->convolution_dimension_numbers().input_spatial_dimensions(1)); in TEST_F()
459 dnums.input_spatial_dimensions(1), in TEST_F()
460 new_conv->convolution_dimension_numbers().input_spatial_dimensions(0)); in TEST_F()
Dhlo_cost_analysis.cc478 dnums.input_spatial_dimensions(spatial_dimension); in HandleConvolution()
Dshape_inference.cc1645 std::copy(dnums.input_spatial_dimensions().begin(), in InferConvolveShape()
1646 dnums.input_spatial_dimensions().end(), input_dnums.begin() + 2); in InferConvolveShape()
1696 input_spatial_dims[i] = lhs.dimensions(dnums.input_spatial_dimensions(i)); in InferConvolveShape()
Dhlo_instruction.cc3054 std::vector<string> lhs_dims(2 + dnums.input_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString()
3057 for (int64 i = 0; i < dnums.input_spatial_dimensions().size(); ++i) { in ConvolutionDimensionNumbersToString()
3058 lhs_dims[dnums.input_spatial_dimensions(i)] = StrCat(i); in ConvolutionDimensionNumbersToString()
Dhlo_evaluator_typed_visitor.h1118 const int64 input_spatial_dim = dnums.input_spatial_dimensions(ki); in HandleConvolution()
Dalgebraic_simplifier.cc3428 const auto& p = padding.dimensions(dnums.input_spatial_dimensions(dim)); in FoldConvInputPad()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc64 new_input_dim_order[i + 1] = dnums.input_spatial_dimensions(i); in Run()
66 input->shape().dimensions(dnums.input_spatial_dimensions(i)); in Run()
Dir_emission_utils.cc93 if (dnums.input_spatial_dimensions(i) != i + 1) { in PotentiallyImplementedAsEigenConvolution()
Dir_emitter.cc1033 lhs->shape().dimensions(dnums.input_spatial_dimensions(i)), in EmitElementalConvolution()
1060 input_multi_index[dnums.input_spatial_dimensions(i)] = input_spatial[i]; in EmitElementalConvolution()
1127 input_shape.dimensions(dnums.input_spatial_dimensions(0)); in HandleConvolution()
1131 : input_shape.dimensions(dnums.input_spatial_dimensions(1)); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc177 std::vector<int64> input_spatial_dimensions; in MakeSpatialPadding() local
184 input_spatial_dimensions.push_back(input_size[dim]); in MakeSpatialPadding()
188 return MakePadding(input_spatial_dimensions, kernel_size_spatial_dimensions, in MakeSpatialPadding()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1669 dimension_numbers.input_spatial_dimensions.extend(range(2, 2 + nd))
1723 dimension_numbers.input_spatial_dimensions.extend(
1940 self.input_spatial_dimensions = []
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py119 dnums.input_spatial_dimensions.extend(range(2, 2 + num_spatial_dims))
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc496 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(0)); in ConvArray4DGeneralDimensionsDilated()
498 lhs_literal.shape().dimensions(dnums.input_spatial_dimensions(1)); in ConvArray4DGeneralDimensionsDilated()
Dxla_data.proto488 repeated int64 input_spatial_dimensions = 11; field
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1090 dimension_numbers.input_spatial_dimensions())); in VerifyConvolution()
1136 lhs_shape.dimensions(dimension_numbers.input_spatial_dimensions(i)); in ConvWithGeneralDimensions()
2681 dnum.input_spatial_dimensions(0), dnum.input_spatial_dimensions(1)}) in Validate()
2687 dnum.input_spatial_dimensions(0), dnum.input_spatial_dimensions(1)); in Validate()