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

Function TFE_OpSetDevice

tensorflow/c/eager/c_api.cc:629–631  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

627void TFE_DeleteOp(TFE_Op* op) { delete op; }
628
629void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) {
630 status->status = op->operation.SetDeviceName(device_name);
631}
632
633const char* TFE_OpGetDevice(TFE_Op* op, TF_Status* status) {
634 tensorflow::Device* device = (op->operation.Device() == nullptr)

Callers 15

TFE_Py_ExecuteCancelableFunction · 0.85
ReadVariableOpFunction · 0.85
TFE_Py_FastPathExecute_CFunction · 0.85
EvaluateOperationFunction · 0.85
Java_org_tensorflow_EagerOperationBuilder_setDeviceFunction · 0.85
createTFEQueueFunction · 0.85
createTFEEnqueueFunction · 0.85
createTFEDequeueFunction · 0.85
TestRemoteExecuteFunction · 0.85
TestRemoteExecuteSilentCopiesFunction · 0.85
TestRemoteExecuteDeleteContextWithOutstandingRPCFunction · 0.85
CheckRemoteMatMulExecutesOKFunction · 0.85

Calls 1

SetDeviceNameMethod · 0.80

Tested by 10

TestRemoteExecuteFunction · 0.68
TestRemoteExecuteSilentCopiesFunction · 0.68
TestRemoteExecuteDeleteContextWithOutstandingRPCFunction · 0.68
CheckRemoteMatMulExecutesOKFunction · 0.68
TensorHandleSilentCopyFunction · 0.68
TensorHandleSilentCopyLocalFunction · 0.68
SetAndGetOpDevicesFunction · 0.68
TESTFunction · 0.68
Execute_MatMul_CPU_Runtime_ErrorFunction · 0.68
ExecuteWithProfilingFunction · 0.68