Searched refs:RngGeneratorKey (Results 1 – 2 of 2) sorted by relevance
41 struct RngGeneratorKey { struct48 friend H AbslHashValue(H h, const RngGeneratorKey& c) { in AbslHashValue() argument53 bool operator==(const RngGeneratorKey& o) const {67 absl::flat_hash_map<RngGeneratorKey, HloComputation*> computation_cache_;
59 RngGeneratorKey cache_key{data_shape, state_shape, algorithm, module}; in GetGeneratorComputation()