Home
last modified time | relevance | path

Searched defs:tensor_device_address (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Doutput_actor.cc354 …auto tensor_device_address = device_context->device_res_manager_->CreateDeviceAddress(kernel_tenso… in CreateOutputTensor() local
390 auto tensor_device_address = std::dynamic_pointer_cast<DeviceTensor>(tensor->device_address()); in UpdateOutputDeviceAddress() local
Ddata_source_actor.cc326 …auto tensor_device_address = std::dynamic_pointer_cast<DeviceTensor>(host_tensor->device_address()… in OnMemoryAllocFinish() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc198 const device::DeviceAddressPtr &tensor_device_address) { in NeedDiscardTensorProperties()