Searched defs:host_tensor (Results 1 – 5 of 5) sorted by relevance
24 void *TensorRTAllocator::MallocDeviceMem(const mindspore::MSTensor &host_tensor, size_t size) { in MallocDeviceMem()84 int TensorRTAllocator::SyncMemInHostAndDevice(mindspore::MSTensor host_tensor, const std::string &d… in SyncMemInHostAndDevice()
140 Status DeviceTensor::DataPop_(std::shared_ptr<Tensor> *host_tensor) { in DataPop_()
273 auto &host_tensor = host_tensors[i]; in OnMemoryAllocFinish() local
31 void SyncTensorData(const TensorPtr &host_tensor, const DeviceTensorPtr &device_tensor, const AnfNo… in SyncTensorData()
173 auto host_tensor = host_tensor_value->cast<std::shared_ptr<tensor::Tensor>>(); in BindAndInitCacheTensor() local