Home
last modified time | relevance | path

Searched defs:kernel_dim (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorConvolution.h34 const Index kernel_dim = kernel_dims[i]; in IndexMapper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dsharding_propagation.cc137 int64_t kernel_dim = in IsConvolutionKernelSmall() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc760 int kernel_dim = descriptor.dnums.kernel_spatial_dimensions(index); in GetConvDescriptor() local
H A Dir_emitter_unnested.cc1028 int kernel_dim = descriptor.dnums.kernel_spatial_dimensions(index); in EmitConvolutionThunk() local