Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for is_gpu_ (0.29 sec)

  1. tensorflow/compiler/jit/device_util.h

      const int kWordSize = 64;
    };
    
    // Caches some miscellaneous information about TF devices.  Thread compatible.
    class DeviceInfoCache {
     public:
      bool IsGpu(DeviceId device) const { return is_gpu_[device.id()]; }
      bool IsCpu(DeviceId device) const { return is_cpu_[device.id()]; }
    
      absl::string_view GetNameFor(DeviceId device) const {
        return names_[device.id()];
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 17:18:31 UTC 2024
    - 7.1K bytes
    - Viewed (0)
Back to top