Home
last modified time | relevance | path

Searched refs:kernel_spatial_dim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc77 int64 kernel_spatial_dim = dim_nums.kernel_spatial_dimensions(i); in ExpandInstruction() local
83 removed_kernel_dimensions.push_back(kernel_spatial_dim); in ExpandInstruction()
88 new_dim_nums.add_kernel_spatial_dimensions(kernel_spatial_dim); in ExpandInstruction()
Dhlo_cost_analysis.cc600 const int64 kernel_spatial_dim = in HandleConvolution() local
602 const int64 kernel_limit = rhs_shape.dimensions(kernel_spatial_dim); in HandleConvolution()
Ddynamic_padder.cc952 int64 kernel_spatial_dim = in RewriteDynamicConvolutionKernelGrad() local
964 custom_call_conv->mutable_operand(1), {}, kernel_spatial_dim); in RewriteDynamicConvolutionKernelGrad()
966 gradients = PadWithScalar(gradients, kernel_spatial_dim, in RewriteDynamicConvolutionKernelGrad()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc308 auto kernel_spatial_dim = in matchAndRewrite() local
317 (*kernel_spatial_dim++).getZExtValue() != i) in matchAndRewrite()