Home
last modified time | relevance | path

Searched refs:CudaTensorParam (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_allocator.h27 struct CudaTensorParam { struct
57 std::map<std::string, CudaTensorParam> cuda_tensor_map_;
Dtensorrt_allocator.cc100 CudaTensorParam &current_cuda_tensor = cuda_tensor_map_.find(device_tensor_name)->second; in SyncMemInHostAndDevice()