Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetCustomDeviceOpHandler (0.17 sec)

  1. tensorflow/c/eager/immediate_execution_context.h

      // Experimental Custom Device.
      //===--------------------------------------------------------------------===//
      virtual CustomDeviceOpHandler& GetCustomDeviceOpHandler() = 0;
    
      // Returns whether `device_name` is registered as a custom device.
      virtual bool IsCustomDevice(const string& device_name) = 0;
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

        TFE_CustomDeviceTensorHandleMethods methods, TF_Status* status) {
      tensorflow::ImmediateExecutionContext* context = tensorflow::unwrap(ctx);
      tensorflow::CustomDevice* device = nullptr;
      if (!context->GetCustomDeviceOpHandler().FindCustomDeviceFromName(device_name,
                                                                        &device)) {
        methods.deallocator(data);
        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)
Back to top