Home
last modified time | relevance | path

Searched refs:output_spatial_dim (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc76 int64 output_spatial_dim = dim_nums.output_spatial_dimensions(i); in ExpandInstruction() local
82 removed_output_dimensions.push_back(output_spatial_dim); in ExpandInstruction()
87 new_dim_nums.add_output_spatial_dimensions(output_spatial_dim); in ExpandInstruction()
Dhlo_cost_analysis.cc606 const int64 output_spatial_dim = in HandleConvolution() local
608 const int64 output_limit = result_shape.dimensions(output_spatial_dim); in HandleConvolution()
Ddynamic_dimension_inference.cc709 int64 output_spatial_dim = in HandleDynamicConvolutionForward() local
718 parent_->SetDynamicSize(hlo, {}, output_spatial_dim, in HandleDynamicConvolutionForward()
Dspace_to_batch_converter.cc2674 const int64 output_spatial_dim = dim_numbers.output_spatial_dimensions( in PerformSpaceToBatchOnConvolution() local
2677 convolution->shape().dimensions(output_spatial_dim); in PerformSpaceToBatchOnConvolution()
2699 .dimensions(output_spatial_dim) in PerformSpaceToBatchOnConvolution()
Ddynamic_padder.cc975 int64 output_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
980 custom_call_conv->shape().dimensions(output_spatial_dim), in RewriteDynamicConvolutionKernelGrad()
Dhlo_evaluator_typed_visitor.h1181 const int64 output_spatial_dim = dnums.output_spatial_dimensions(ki);
1187 out_index[output_spatial_dim] * window_dim.stride() -
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc310 auto output_spatial_dim = in matchAndRewrite() local
316 (*output_spatial_dim++).getZExtValue() != dim || in matchAndRewrite()