Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/eager/c_api_experimental.h

                                                        TF_Status* status);
    
    // Returns whether `device_name` maps to a registered custom device.
    TF_CAPI_EXPORT extern bool TFE_IsCustomDevice(TFE_Context* ctx,
                                                  const char* device_name);
    
    // Struct to be filled in to define a custom device tensor handle. Fields are
    // required except where indicated.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

      TF_SetStatus(status, TF_UNIMPLEMENTED,
                   "This custom device does not support packing tensors.");
      return nullptr;
    }
    }  // 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,
    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