- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TFE_OpSetDevice (0.34 sec)
-
tensorflow/c/eager/c_api.cc
return tensorflow::unwrap(op)->Name().c_str(); } TFE_Context* TFE_OpGetContext(const TFE_Op* op, TF_Status* status) { return tensorflow::wrap(tensorflow::unwrap(op)->GetContext()); } void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { status->status = tensorflow::unwrap(op)->SetDeviceName(device_name); } const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)