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