Home
last modified time | relevance | path

Searched defs:input_spatial_dim (Results 1 – 5 of 5) 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
H A Ddynamic_padder.cc976 int64_t input_spatial_dim = in RewriteDynamicConvolutionInputGrad() local
1046 int64_t input_spatial_dim = in RewriteDynamicConvolutionForward() local
1107 int64_t input_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
H A Dhlo_cost_analysis.cc630 const int64_t input_spatial_dim = in GetConvolutionFlops() local
H A Ddynamic_dimension_inference.cc755 int64_t input_spatial_dim = in HandleDynamicConvolutionForward() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc476 auto input_spatial_dim = dnums.getInputSpatialDimensions(); in IsSamePadding() local