Searched refs:resource_device_ (Results 1 – 2 of 2) sorted by relevance
55 resource_device_(GetResourceDevice(t, ctx)), in TensorHandle()70 resource_device_(resource_device), in TensorHandle()77 DCHECK(dtype == DT_RESOURCE ? resource_device_ != nullptr in TensorHandle()78 : resource_device_ == nullptr); in TensorHandle()90 resource_device_(resource_device), in TensorHandle()99 DCHECK(dtype == DT_RESOURCE ? resource_device_ != nullptr in TensorHandle()100 : resource_device_ == nullptr); in TensorHandle()108 resource_device_(nullptr), in TensorHandle()
86 tensorflow::Device* resource_device() const { return resource_device_; } in resource_device()168 tensorflow::Device* const resource_device_; variable