Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 2 of 2 for TFE_IsCustomDevice (0.07 seconds)

  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.
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Click Count (1)
  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,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Nov 07 05:55:21 GMT 2025
    - 43.9K bytes
    - Click Count (0)
Back to Top