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