Searched refs:GetShardingSpec (Results 1 – 2 of 2) sorted by relevance
170 const ShardingSpec& GetShardingSpec() const { return sharding_spec_; } in GetShardingSpec() function
413 &ShardedDeviceArray::GetShardingSpec) in BuildPmapSubmodule()