| 54 | } |
| 55 | |
| 56 | port::Status TfAllocatorAdapter::Deallocate(int device_ordinal, |
| 57 | DeviceMemoryBase mem) { |
| 58 | wrapped_->DeallocateRaw(mem.opaque()); |
| 59 | return port::Status::OK(); |
| 60 | } |
| 61 | |
| 62 | port::StatusOr<Stream *> TfAllocatorAdapter::GetStream(int device_ordinal) { |
| 63 | CHECK_EQ(stream_->parent()->device_ordinal(), device_ordinal); |
no test coverage detected