Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for is_gpu_ (0.1 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)
  2. tensorflow/compiler/jit/device_util.cc

      DeviceType* device_type = id_to_device_type_.back().get();
      TF_RETURN_IF_ERROR(DeviceNameToDeviceType(names_.back(), device_type));
    
      is_cpu_.push_back(device_type->type_string() == DEVICE_CPU);
      is_gpu_.push_back(device_type->type_string() == DEVICE_GPU);
    
      name_to_id_.emplace(string(name), DeviceId(new_id));
    
      const XlaOpRegistry::DeviceRegistration* compilation_device;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.8K bytes
    - Viewed (0)
Back to top