Searched defs:device_tensor_name (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/ |
| D | tensorrt_allocator.cc | 82 …ator::SyncMemHostToDevice(const tensor::Tensor &host_tensor, const std::string &device_tensor_name, in SyncMemHostToDevice()
88 …TAllocator::SyncMemDeviceToHost(tensor::Tensor *host_tensor, const std::string &device_tensor_name, in SyncMemDeviceToHost()
125 …cator::SyncMemDeviceToHost(void *dst_data, size_t data_size, const std::string &device_tensor_name, in SyncMemDeviceToHost()
157 …locator::SyncMemInHostAndDevice(tensor::Tensor *host_tensor, const std::string &device_tensor_name, in SyncMemInHostAndDevice()
188 …ensorRTAllocator::SyncMemInHostAndDevice(void *host_data, const std::string &device_tensor_name, s… in SyncMemInHostAndDevice()
|