Searched refs:UnderlyingDevice (Results 1 – 4 of 4) sorted by relevance
39 const DeviceBase* UnderlyingDevice() const override { in UnderlyingDevice() function40 return underlying_device_->UnderlyingDevice(); in UnderlyingDevice()42 DeviceBase* UnderlyingDevice() override { in UnderlyingDevice() function43 return underlying_device_->UnderlyingDevice(); in UnderlyingDevice()
214 virtual DeviceBase* UnderlyingDevice() { return this; } in UnderlyingDevice() function215 virtual const DeviceBase* UnderlyingDevice() const { return this; } in UnderlyingDevice() function
131 dynamic_cast<XlaDevice*>(ctx->device()->UnderlyingDevice()); in RunTransfer()
173 XlaDevice* xla_device = dynamic_cast<XlaDevice*>(device->UnderlyingDevice()); in GetMetadataFromDevice()