Searched defs:host_tensor (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/ |
| D | tensorrt_allocator.cc | 25 void *TensorRTAllocator::MallocDeviceMem(const TensorInfo &host_tensor, size_t size) { in MallocDeviceMem()
82 int TensorRTAllocator::SyncMemHostToDevice(const tensor::Tensor &host_tensor, const std::string &de… in SyncMemHostToDevice()
88 int TensorRTAllocator::SyncMemDeviceToHost(tensor::Tensor *host_tensor, const std::string &device_t… in SyncMemDeviceToHost()
157 int TensorRTAllocator::SyncMemInHostAndDevice(tensor::Tensor *host_tensor, const std::string &devic… in SyncMemInHostAndDevice()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/ |
| D | device_tensor_ascend910b.cc | 179 Status DeviceTensorAscend910B::ToHostTensor(std::shared_ptr<Tensor> *host_tensor) { in ToHostTensor()
|
| D | device_tensor.cc | 133 Status DeviceTensor::DataPop_(std::shared_ptr<Tensor> *host_tensor) { in DataPop_()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | data_source_actor.cc | 317 auto &host_tensor = host_tensors[i]; in OnMemoryAllocFinish() local
|
| D | data_prepare_actor.cc | 70 device::StorageInfo GetStorageInfo(const TensorPtr &host_tensor, const DeviceTensorPtr &device_tens… in GetStorageInfo()
109 void SyncTensorData(const TensorPtr &host_tensor, const DeviceTensorPtr &device_tensor, const AnfNo… in SyncTensorData()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | acl_convert.cc | 1109 …Desc *, aclDataBuffer *> AclConverter::ConvertTensorToAclDesc(const tensor::TensorPtr &host_tensor, in ConvertTensorToAclDesc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 179 auto host_tensor = host_tensor_value->cast<std::shared_ptr<tensor::Tensor>>(); in BindAndInitCacheTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | ascend_device_address.cc | 669 auto host_tensor = std::make_shared<tensor::Tensor>(src_device_address->type_id(), host_shape); in SyncDeviceToDeviceWithDiffFormatType() local
|