Home
last modified time | relevance | path

Searched refs:input_spatial_dim (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_4d_expander.cc75 int64_t input_spatial_dim = dim_nums.input_spatial_dimensions(i); in ExpandInstruction() local
78 if (input->shape().dimensions(input_spatial_dim) == 1 && in ExpandInstruction()
81 removed_input_dimensions.push_back(input_spatial_dim); in ExpandInstruction()
86 new_dim_nums.add_input_spatial_dimensions(input_spatial_dim); in ExpandInstruction()
H A Ddynamic_padder.cc976 int64_t input_spatial_dim = in RewriteDynamicConvolutionInputGrad() local
980 custom_call_conv->mutable_operand(1), {}, input_spatial_dim); in RewriteDynamicConvolutionInputGrad()
984 grad = PadWithScalar(grad, input_spatial_dim, operand_dynamic_size, zero); in RewriteDynamicConvolutionInputGrad()
988 custom_call_conv->mutable_operand(0), {input_spatial_dim}, in RewriteDynamicConvolutionInputGrad()
989 {input_spatial_dim + 1}, {1})); in RewriteDynamicConvolutionInputGrad()
1046 int64_t input_spatial_dim = in RewriteDynamicConvolutionForward() local
1050 custom_call_conv->mutable_operand(0), {}, input_spatial_dim); in RewriteDynamicConvolutionForward()
1055 input = PadWithScalar(input, input_spatial_dim, operand_dynamic_size, zero); in RewriteDynamicConvolutionForward()
1107 int64_t input_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
1113 custom_call_conv->mutable_operand(0), {}, input_spatial_dim); in RewriteDynamicConvolutionKernelGrad()
[all …]
H A Dhlo_cost_analysis.cc630 const int64_t input_spatial_dim = in GetConvolutionFlops() local
632 const int64_t input_limit = lhs_shape.dimensions(input_spatial_dim); in GetConvolutionFlops()
H A Dhlo_evaluator_typed_visitor.h999 const int64_t input_spatial_dim = dnums.input_spatial_dimensions(ki);
1028 lhs_spatial_index < lhs_shape.dimensions(input_spatial_dim))) {
1033 lhs_spatial_index * lhs_dim_multipliers[input_spatial_dim];
H A Ddynamic_dimension_inference.cc755 int64_t input_spatial_dim = in HandleDynamicConvolutionForward() local
759 if (dimension == input_spatial_dim) { in HandleDynamicConvolutionForward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc476 auto input_spatial_dim = dnums.getInputSpatialDimensions(); in IsSamePadding() local
485 input_spatial_dim[i]), in IsSamePadding()