Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc77 int64_t kernel_spatial_dim = dim_nums.kernel_spatial_dimensions(i); in ExpandInstruction() local
Dhlo_cost_analysis.cc618 const int64_t kernel_spatial_dim = in GetConvolutionFlops() local
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
Ddynamic_padder.cc1109 int64_t kernel_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc477 auto kernel_spatial_dim = dnums.getKernelSpatialDimensions(); in IsSamePadding() local