Searched refs:ReshardCache (Results 1 – 2 of 2) sorted by relevance
269 struct ReshardCache { struct280 absl::flat_hash_map<std::string, std::unique_ptr<ReshardCache>> argument290 ReshardCache* reshard_cache; argument535 PartitionedHlo::ReshardCache reshard_cache_;
1687 grouped_cache = absl::make_unique<PartitionedHlo::ReshardCache>(); in CreatePerGroupPartitioningState()