| 178 | |
| 179 | const TensorShape& TensorLeaf::shape_impl() const { return tensor_ptr_->shape(); } |
| 180 | Device TensorLeaf::device_impl() const { return tensor_ptr_->device(); } |
| 181 | DataType TensorLeaf::dtype_impl() const { return tensor_ptr_->dtype(); } |
| 182 | cudaStream_t TensorLeaf::stream_hint_impl() const { return tensor_ptr_ ? tensor_ptr_->stream() : nullptr; } |
| 183 |