Searched refs:SupportSpatialPartitioning (Results 1 – 1 of 1) sorted by relevance
321 bool SupportSpatialPartitioning( in SupportSpatialPartitioning() function749 if (!SupportSpatialPartitioning(instruction, computation_map, is_spmd, false, in InferShardingFromUsers()1738 if (!SupportSpatialPartitioning(instruction, computation_map, is_spmd_, in InferShardingFromOperands()