Searched refs:input_spatial_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_4d_expander.cc | 75 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()
|
| D | dynamic_padder.cc | 920 int64_t input_spatial_dim = in RewriteDynamicConvolutionInputGrad() local
924 custom_call_conv->mutable_operand(1), {}, input_spatial_dim); in RewriteDynamicConvolutionInputGrad()
928 grad = PadWithScalar(grad, input_spatial_dim, operand_dynamic_size, zero); in RewriteDynamicConvolutionInputGrad()
931 {input_spatial_dim}, {input_spatial_dim + 1}, {1})); in RewriteDynamicConvolutionInputGrad()
989 int64_t input_spatial_dim = in RewriteDynamicConvolutionForward() local
993 custom_call_conv->mutable_operand(0), {}, input_spatial_dim); in RewriteDynamicConvolutionForward()
998 input = PadWithScalar(input, input_spatial_dim, operand_dynamic_size, zero); in RewriteDynamicConvolutionForward()
1051 int64_t input_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
1057 custom_call_conv->mutable_operand(0), {}, input_spatial_dim); in RewriteDynamicConvolutionKernelGrad()
1059 activations = PadWithScalar(activations, input_spatial_dim, in RewriteDynamicConvolutionKernelGrad()
|
| D | hlo_cost_analysis.cc | 613 const int64_t input_spatial_dim = in HandleConvolution() local
615 const int64_t input_limit = lhs_shape.dimensions(input_spatial_dim); in HandleConvolution()
|
| D | dynamic_dimension_inference.cc | 726 int64_t input_spatial_dim = in HandleDynamicConvolutionForward() local
730 if (dimension == input_spatial_dim) { in HandleDynamicConvolutionForward()
|
| D | hlo_evaluator_typed_visitor.h | 1198 const int64_t input_spatial_dim = dnums.input_spatial_dimensions(ki);
1227 lhs_spatial_index < lhs_shape.dimensions(input_spatial_dim))) {
1232 lhs_spatial_index * lhs_dim_multipliers[input_spatial_dim];
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 215 auto input_spatial_dim = dimension_numbers.input_spatial_dimensions().begin(); in HasCanonicalDimensionNumbers() local
223 if ((*input_spatial_dim++).getZExtValue() != dim || in HasCanonicalDimensionNumbers()
|