Home
last modified time | relevance | path

Searched defs:kernel_spatial_dim (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_4d_expander.cc77 int64_t kernel_spatial_dim = dim_nums.kernel_spatial_dimensions(i); in ExpandInstruction() local
H A Dspace_to_batch_converter.cc385 const int64_t kernel_spatial_dim = dim_numbers.kernel_spatial_dimensions( in IsForwardWindowDilatedConv() local
3548 const int64_t kernel_spatial_dim = dim_numbers.kernel_spatial_dimensions( in GetConvolutionDetails() local
H A Dhlo_cost_analysis.cc618 const int64_t kernel_spatial_dim = in GetConvolutionFlops() local
H A Ddynamic_padder.cc1109 int64_t kernel_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc477 auto kernel_spatial_dim = dnums.getKernelSpatialDimensions(); in IsSamePadding() local