Searched defs:lhs_dimension (Results 1 – 1 of 1) sorted by relevance
323 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local360 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local631 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local659 int64_t lhs_dimension = dnums.input_spatial_dimensions(i); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local