Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for list_devices (0.11 sec)

  1. tensorflow/c/eager/immediate_execution_context.h

      // Runtime. This is necessary to decouple runtime-dependent
      // code that is layered on top of the runtime.
      virtual bool UsesTFRT() = 0;
    
      // List attributes of available devices
      virtual void ListDevices(std::vector<DeviceAttributes>* devices) = 0;
    
      // Add `devices` into context's device manager. Context's device manager
      // will take ownership and maintain devices' lifetime.
      virtual absl::Status AddDevices(
    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

      tensorflow::unwrap(ctx)->Release();
    }
    
    TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) {
      TF_DeviceList* l = new TF_DeviceList;
      tensorflow::unwrap(ctx)->ListDevices(&l->response);
      return l;
    }
    
    void TFE_ContextClearCaches(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->ClearCachesAndThreadExecutors();
    }
    
    // Set server_def on the context, possibly updating it.
    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