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 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 | 822 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionInputGrad() local
825 custom_call_conv->mutable_operand(1), {}, input_spatial_dim); in RewriteDynamicConvolutionInputGrad()
829 grad = PadWithScalar(grad, input_spatial_dim, operand_dynamic_size, zero); in RewriteDynamicConvolutionInputGrad()
832 {input_spatial_dim}, {input_spatial_dim + 1}, {1})); in RewriteDynamicConvolutionInputGrad()
890 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionForward() local
893 custom_call_conv->mutable_operand(0), {}, input_spatial_dim); in RewriteDynamicConvolutionForward()
898 input = PadWithScalar(input, input_spatial_dim, operand_dynamic_size, zero); in RewriteDynamicConvolutionForward()
951 int64 input_spatial_dim = dnums.input_spatial_dimensions(spatial_dim_index); in RewriteDynamicConvolutionKernelGrad() local
956 custom_call_conv->mutable_operand(0), {}, input_spatial_dim); in RewriteDynamicConvolutionKernelGrad()
958 activations = PadWithScalar(activations, input_spatial_dim, in RewriteDynamicConvolutionKernelGrad()
|
| D | hlo_cost_analysis.cc | 612 const int64 input_spatial_dim = in HandleConvolution() local
614 const int64 input_limit = lhs_shape.dimensions(input_spatial_dim); in HandleConvolution()
|
| D | dynamic_dimension_inference.cc | 707 int64 input_spatial_dim = in HandleDynamicConvolutionForward() local
711 if (dimension == input_spatial_dim) { in HandleDynamicConvolutionForward()
|
| D | hlo_evaluator_typed_visitor.h | 1180 const int64 input_spatial_dim = dnums.input_spatial_dimensions(ki);
1208 lhs_spatial_index < lhs_shape.dimensions(input_spatial_dim))) {
1213 lhs_spatial_index * lhs_dim_multipliers[input_spatial_dim];
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 306 auto input_spatial_dim = in matchAndRewrite() local
315 if ((*input_spatial_dim++).getZExtValue() != dim || in matchAndRewrite()
|