Searched refs:f16_allowlist_ (Results 1 – 1 of 1) sorted by relevance
1010 gtl::FlatSet<string> f16_allowlist_; member in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl1268 f16_allowlist_ = mp_lists->AllowList(); in Optimize()1272 TF_RETURN_IF_ERROR(ValidateLists(f16_allowlist_, f16_denylist_, in Optimize()1543 if (f16_allowlist_.count(root.node->op()) || force_allow) { in AddAllowlistOps()1615 if (!ShouldProcess(*root.node) || !f16_allowlist_.count(root.node->op())) { in AddClearAndInferToAllowIfBetweenAllow()1624 !f16_allowlist_.count(item.node->op()) && in AddClearAndInferToAllowIfBetweenAllow()1642 !f16_allowlist_.count(root.node->op())) { in AddClearAndInferToAllowIfBetweenAllow()1711 if (f16_allowlist_.count(root.node->op()) && allow_set->count(root_idx) && in RemoveAllowsetWithFp32()