Searched refs:control_overrides (Results 1 – 1 of 1) sorted by relevance
306 const absl::flat_hash_map<string, std::vector<string>>& control_overrides() in control_overrides() function in tensorflow::grappler::__anon35816ea30111::FunctionOptimizerContext381 const std::vector<string>& control_overrides) { in AddControlOverrides() argument383 << absl::StrJoin(control_overrides, ", ") << "]"; in AddControlOverrides()385 control_overrides_[func_node.name()].reserve(control_overrides.size()); in AddControlOverrides()386 for (const string& control_override : control_overrides) { in AddControlOverrides()1965 if (!ctx.control_overrides().empty()) { in RestoreGraphInvariants()1976 auto overrides = ctx.control_overrides().find(input_tensor.node()); in RestoreGraphInvariants()1977 if (overrides != ctx.control_overrides().end()) { in RestoreGraphInvariants()