Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_OpGetDevice (0.15 sec)

  1. tensorflow/c/eager/custom_device_testutil.cc

    void LoggingDeviceExecute(const TFE_Op* original_op, int* num_outputs,
                              TFE_TensorHandle** outputs, TF_Status* s,
                              void* device_info) {
      const char* requested_placement = TFE_OpGetDevice(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
    
      LoggingDevice* dev = reinterpret_cast<LoggingDevice*>(device_info);
      if (dev->strict_scope_placement && *requested_placement == '\0') {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Mar 03 20:47:31 UTC 2021
    - 8.3K bytes
    - Viewed (0)
Back to top