Home
last modified time | relevance | path

Searched defs:current_cuda_tensor (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_allocator.cc96 CudaTensorParam &current_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemDeviceToHost() local
136 CudaTensorParam &current_cuda_tensor = it->second; in SyncMemDeviceToHost() local
165 CudaTensorParam &current_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemInHostAndDevice() local
194 CudaTensorParam &current_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemInHostAndDevice() local