Searched refs:LimitToAssignedDevice (Results 1 – 3 of 3) sorted by relevance
126 TF_RETURN_IF_ERROR(colocation_graph->LimitToAssignedDevice(*node)); in AssignAndLog()189 TF_RETURN_IF_ERROR(colocation_graph.LimitToAssignedDevice(*node)); in Run()
237 Status LimitToAssignedDevice(const Node& node);
1094 Status ColocationGraph::LimitToAssignedDevice(const Node& node) { in LimitToAssignedDevice() function in tensorflow::ColocationGraph