Home
last modified time | relevance | path

Searched defs:input_device_address (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/
Dgpu_convert_to_dynamic_shape_gpu_kernel.h39 T *input_device_address = GetDeviceAddress<T>(inputs, 0); in Launch() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/acl/
Dacl_lite_kernel.cc105 auto input_device_address = input->device_data(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dmatrix_set_diag_v3_gpu_kernel.cc100 auto input_device_address = reinterpret_cast<T *>(input->device_ptr()); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/
Dpyboost_utils.cc229 …auto input_device_address = std::dynamic_pointer_cast<device::DeviceAddress>(input->device_address… in CreateOutputTensor() local
285 const device::DeviceAddressPtrList &input_device_address) { in GetKernelTensorFromAddress()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Ddynamic_shape_helper.cc205 auto input_device_address = reinterpret_cast<std::vector<device::DeviceAddress *> *>(args); in GetDependValueTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/
Dsingle_op_session.cc374 auto input_device_address = input.device_address(); in InitInputOutputData() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dkernel_actor.cc183 const auto &input_device_address = in InitInputInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_subgraph.cc783 auto input_device_address = inputs[i].device_address(); in PreExecute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/
Dforward.cc1011 …auto input_device_address = std::dynamic_pointer_cast<device::DeviceAddress>(input_tensor->device_… in CreateViewOutputTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc843 …auto input_device_address = AnfAlgo::GetMutableOutputAddr(input_with_index.first, input_with_index… in CreateOutputDeviceAddress() local