Searched refs:output_spatial_dim (Results 1 – 7 of 7) sorted by relevance
76 int64 output_spatial_dim = dim_nums.output_spatial_dimensions(i); in ExpandInstruction() local82 removed_output_dimensions.push_back(output_spatial_dim); in ExpandInstruction()87 new_dim_nums.add_output_spatial_dimensions(output_spatial_dim); in ExpandInstruction()
606 const int64 output_spatial_dim = in HandleConvolution() local608 const int64 output_limit = result_shape.dimensions(output_spatial_dim); in HandleConvolution()
709 int64 output_spatial_dim = in HandleDynamicConvolutionForward() local718 parent_->SetDynamicSize(hlo, {}, output_spatial_dim, in HandleDynamicConvolutionForward()
2674 const int64 output_spatial_dim = dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution() local2677 convolution->shape().dimensions(output_spatial_dim); in PerformSpaceToBatchOnConvolution()2699 .dimensions(output_spatial_dim) in PerformSpaceToBatchOnConvolution()
975 int64 output_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local980 custom_call_conv->shape().dimensions(output_spatial_dim), in RewriteDynamicConvolutionKernelGrad()
1181 const int64 output_spatial_dim = dnums.output_spatial_dimensions(ki);1187 out_index[output_spatial_dim] * window_dim.stride() -
310 auto output_spatial_dim = in matchAndRewrite() local316 (*output_spatial_dim++).getZExtValue() != dim || in matchAndRewrite()