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