MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SetRequestedDevice

Function SetRequestedDevice

tensorflow/c/python_api.cc:52–56  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

50}
51
52void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) {
53 mutex_lock l(graph->mu);
54 op->node.set_requested_device(device);
55 RecordMutation(graph, *op, "setting device");
56}
57
58void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst,
59 TF_Status* status) {

Callers

nothing calls this directly

Calls 2

RecordMutationFunction · 0.85
set_requested_deviceMethod · 0.80

Tested by

no test coverage detected