Home
last modified time | relevance | path

Searched defs:device_tensor (Results 1 – 25 of 40) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/memory/
Dmemory_swap_actor.cc76 for (const auto device_tensor : device_tensors) { in AllocDeviceContinuousMem() local
107 for (const auto &device_tensor : device_tensors) { in Swap() local
162 for (const auto &device_tensor : device_tensors_to_swap_) { in Run() local
181 for (const auto &device_tensor : device_tensors_to_swap_) { in Run() local
190 for (const auto &device_tensor : device_tensors_to_free_) { in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/
Ddvpp_image_utils.cc112 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAdjustBrightness() local
202 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAdjustContrast() local
292 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAdjustHue() local
382 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAdjustSaturation() local
473 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAdjustSharpness() local
590 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAffine() local
708 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppAutoContrast() local
870 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppConvertColor() local
981 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppCrop() local
1100 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in DvppErase() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/
Dnpu_map_job.cc58 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in Run() local
86 std::shared_ptr<DeviceTensorAscend910B> device_tensor = nullptr; in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Doutput_actor.cc196 …const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(device_tensor_store_key.second, 0, false… in RunOpControl() local
330 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor() local
378 auto device_tensor = output_device_tensors_[i]; in UpdateOutputDeviceAddress() local
Ddata_prepare_actor.cc70 …ice::StorageInfo GetStorageInfo(const TensorPtr &host_tensor, const DeviceTensorPtr &device_tensor, in GetStorageInfo()
103 device::tracker::MemType mem_type, const DeviceTensorPtr &device_tensor) { in UpdateTracker()
109 void SyncTensorData(const TensorPtr &host_tensor, const DeviceTensorPtr &device_tensor, const AnfNo… in SyncTensorData()
200 const auto &device_tensor = AnfAlgo::GetPrevNodeMutableOutputAddr(node, i, false); in FetchContinuousMemoryInfo() local
211 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, i, false); in FetchContinuousMemoryInfo() local
733 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(value_node, 0, false); in PrepareDataForDeviceTensorStore() local
941 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, 0, false); in PrepareDataForValueNodeTensor() local
1003 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, index, false); in PrepareDataForControlValueNode() local
1053 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, index, false); in PrepareDataForStringValue() local
1115 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, index, false); in PrepareDataForSequenceAndScalarValue() local
[all …]
Dany_type_kernel_actor.cc152 auto device_tensor = DeviceTensorStore::GetInstance() in FetchInputDeviceTensor() local
236 const auto &device_tensor = device_tensors[index]; in GenerateIDForGraph() local
278 const auto &device_tensor = device_tensors[index]; in InferParameterAbstractForModelGraph() local
614 void FreeMemory(DeviceTensor *device_tensor) { in FreeMemory()
736 [](DeviceTensor *device_tensor) { device_tensor->set_ptr(nullptr); }); in FetchGraphOutput()
Dmemory_manager_actor.cc169 auto &device_tensor = (*alloc_list)[i]; in AllocateBatchMemory() local
292 auto &device_tensor = (*free_list)[i]; in FreeBatchMemory() local
364 void MemoryManagerActor::FreeMemoryByRefCount(DeviceTensor *const device_tensor, const DeviceContex… in FreeMemoryByRefCount()
Ddata_source_actor.cc117 for (auto &device_tensor : kernel_info_->output_address_list()) { in FillDataBuffer() local
318 auto &device_tensor = device_tensors[i]; in OnMemoryAllocFinish() local
Dactor_common.cc362 void UpdateRefCount(DeviceTensor *const device_tensor, bool is_max_ref_count) { in UpdateRefCount()
374 auto device_tensor = AnfAlgo::GetMutableOutputAddr(node, output_idx, false); in UpdateRefCount() local
378 void FreeMemoryByDeviceContext(DeviceTensor *const device_tensor, const DeviceContext *device_conte… in FreeMemoryByDeviceContext()
391 …alueNode(const std::vector<std::weak_ptr<ValueNode>> &held_by_nodes, DeviceTensor *device_tensor) { in FreeMemoryByValueNode()
Dactor_dump.cc189 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(data_kernel, i, false); in DumpDSActor() local
205 …const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(data_node.first, data_node.second, false… in DumpDSActor() local
237 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(kernel, i, false); in DumpKernelActor() local
251 auto &device_tensor = workspace_addresses[i]; in DumpKernelActor() local
351 auto &device_tensor = actor->output(); in DumpCopyActor() local
404 for (const auto &device_tensor : formal_parameter_device_tensor.second) { in DumpFormalParameterDeviceTensor() local
419 for (const auto &device_tensor : ref_node_formal_parameter_device_tensor.second) { in DumpFormalParameterDeviceTensor() local
Dkernel_actor.cc593 const auto device_tensor = memory_free_list_[i]; in SendMemoryFreeReq() local
941 [](auto &device_tensor) { device_tensor->set_need_sync_user_data(true); }); in ExecuteLaunchKernelTask()
1236 [](auto &device_tensor) { device_tensor->set_need_sync_user_data(true); }); in PostLaunchKernel()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/
Dembedding_cache.cc64 …us EmbeddingCache::Init(mindspore::MSTensor host_cache_tensor, mindspore::MSTensor device_tensor) { in Init()
126 mindspore::MSTensor device_tensor) { in Init()
Dembedding_cache_manager.cc100 auto device_tensor = kernel->inputs()[0]; in InitCacheKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/
Dembedding_cache.cc63 …us EmbeddingCache::Init(mindspore::MSTensor host_cache_tensor, mindspore::MSTensor device_tensor) { in Init()
125 mindspore::MSTensor device_tensor) { in Init()
Dembedding_cache_manager.cc100 auto device_tensor = kernel->inputs()[0]; in InitCacheKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/
Dgather_actor.cc140 const auto &device_tensor = input_device_tensors_[i]; in GatherInput() local
212 for (auto &device_tensor : output->device_tensors_) { in BuildOutput() local
Dcontrol_actor.cc57 [](const auto &device_tensor) { return device_tensor.second; }); in GetAllDeviceTensors()
68 for (auto &device_tensor : op_real_parameter.device_tensors_) { in GetAllDeviceTensors() local
412 for (auto &device_tensor : ref_node_formal_parameter_device_tensors_[formal_parameter_position]) { in UpdateOutputData() local
565 … const std::vector<DeviceTensor *> &addr_list, DeviceTensor **device_tensor) { in MergeDeviceAddress()
667 DeviceTensor **device_tensor) { in MergeEmptyAddressDeviceAddress()
Dswitch_actor.cc73 DeviceTensor *device_tensor = input_device_tensors_[0]; in GetIndex() local
Dcondition_gather_actor.cc145 … auto device_tensor = DeviceTensorStore::GetInstance().Fetch(device_tensor_store_key.second.get(), in FetchInput() local
Dcondition_switch_actor.cc181 … auto device_tensor = DeviceTensorStore::GetInstance().Fetch(device_tensor_store_key.second.get(), in FetchInput() local
Dentrance_actor.cc127 for (const auto &device_tensor : device_tensors) { in FetchInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dscheduler_helper.cc123 const DeviceTensorPtr &device_tensor) { in UpdateDataArrowRefCount()
248 …edulerHelper::AddDeviceTensorStore(const AnfNode *anf_node, const DeviceTensorPtr &device_tensor) { in AddDeviceTensorStore()
398 auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_kernel, from_output_index, false); in AddDataArrow() local
437 auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_kernel, from_output_index, false); in AddResultArrow() local
599 auto device_tensor = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in AddFormalParameterDeviceTensor() local
621 auto device_tensor = in ConvertDataArrowToControlArrow() local
Ddevice_tensor_store.h97 for (const auto &device_tensor : iter->second) { in Fetch() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Ddevice_tensor_ascend910b.cc141 void *device_tensor = nullptr; in CreateDeviceTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/
Dsend_actor.cc177 for (const auto &device_tensor : workspace_device_tensors_) { in FindDeviceTensorNeedsFree() local

12