Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for RegisterCustomDevice (0.19 sec)

  1. tensorflow/c/eager/custom_device_testutil.cc

      device->executed_flag = executed_flag;
      device->device_name = name;
      device->underlying_device = "/job:localhost/replica:0/task:0/device:CPU:0";
      device->strict_scope_placement = strict_scope_placement;
      TFE_RegisterCustomDevice(context, custom_device, name, device, status);
    }
    
    TFE_TensorHandle* UnpackTensorHandle(TFE_TensorHandle* logged_tensor_handle,
                                         TF_Status* status) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

    }
    }  // namespace
    
    extern "C" {
    
    bool TFE_IsCustomDevice(TFE_Context* ctx, const char* device_name) {
      return tensorflow::unwrap(ctx)->IsCustomDevice(device_name);
    }
    
    void TFE_RegisterCustomDevice(TFE_Context* ctx, TFE_CustomDevice device,
                                  const char* device_name, void* device_info,
                                  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)
Back to top