Home
last modified time | relevance | path

Searched refs:allow_set (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc977 void AddAllowlistOps(absl::flat_hash_set<int>* allow_set) const;
978 void RemoveAllowsetWithFp32(absl::flat_hash_set<int>* allow_set) const;
983 absl::flat_hash_set<int>* allow_set,
987 absl::flat_hash_set<int>* allow_set) const;
989 absl::flat_hash_set<int>* allow_set) const;
991 absl::flat_hash_set<int>* allow_set) const;
993 absl::flat_hash_set<int>* allow_set) const;
996 Status ChangeTypeAttrsAndAddCasts(const absl::flat_hash_set<int>& allow_set);
1350 absl::flat_hash_set<int> allow_set; in Optimize() local
1352 AddAllowlistOps(&allow_set); in Optimize()
[all …]