Searched defs:rhs_dimension (Results 1 – 1 of 1) sorted by relevance
324 int64 rhs_dimension = dnums.kernel_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local361 int64 rhs_dimension = dnums.kernel_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local632 int64 rhs_dimension = dnums.kernel_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local