Searched refs:AddNodeControl (Results 1 – 2 of 2) sorted by relevance
64 NodeDef* AutoParallel::AddNodeControl(const string& name, in AddNodeControl() function in tensorflow::grappler::AutoParallel253 auto control = AddNodeControl(name_control, fetches, graph); in BuildGraph()256 AddNodeControl(fetch, {control->name()}, graph); in BuildGraph()
55 NodeDef* AddNodeControl(const string& name, const std::set<string>& deps,