Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for DLDevice (0.04 sec)

  1. tensorflow/c/eager/dlpack.cc

              DataType_Name(static_cast<DataType>(data_type)),
              " is not supported by dlpack");
          break;
      }
      return dtype;
    }
    
    // Gets DLPack's DLDevice from eager tensor handle.
    DLDevice GetDlContext(TFE_TensorHandle* h, TF_Status* status) {
      DLDevice ctx;
      const char* device_name =
          tensorflow::unwrap(h)->BackingDeviceName(&status->status);
      DeviceNameUtils::ParsedName parsed_name;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
Back to top