- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TFE_OpGetDevice (0.23 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.cc
void ParallelDeviceExecute(const TFE_Op* original_op, int* num_outputs, TFE_TensorHandle** outputs, TF_Status* status, void* device_info) { const char* requested_placement = TFE_OpGetDevice(original_op, status); if (*requested_placement == '\0') { TF_SetStatus( status, TF_INTERNAL, "Ops must be placed on the parallel device explicitly, or their inputs "
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} 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) { return tensorflow::unwrap(op)->DeviceName().c_str(); } void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)