Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc254 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareDataForHostTensorQueue() local
567 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareHostTensorQueueForControlNode() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc182 auto tensor_address = tensor->device_address(); in LoadInputData() local
Dgpu_session.cc344 … auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in LoadInputData() local
Dascend_session.cc330 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in TensorNeedSync() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc326 auto tensor_address = tensor->device_address(); in BindInputTensorAddressPtr() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc936 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignExtraStaticMem() local
1402 auto tensor_address = tensor->device_address(); in InitGraphInputTensors() local