Searched refs:sharding_spec_ (Results 1 – 1 of 1) sorted by relevance
166 sharding_spec_(std::move(sharding_spec)), in ShardedDeviceArray()170 const ShardingSpec& GetShardingSpec() const { return sharding_spec_; } in GetShardingSpec()177 ShardingSpec sharding_spec_; variable