Home
last modified time | relevance | path

Searched refs:input2_addr_ (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_mul.cc71 input2_addr_ = AllocDeviceMem(input_size); in ObtainMulInputsAddr()
73 inputs_addr_.push_back(input2_addr_); in ObtainMulInputsAddr()
78 if (input2_addr_ != nullptr) { in FreeInputDeviceMemory()
79 FreeDeviceMem(input2_addr_); in FreeInputDeviceMemory()
80 input2_addr_ = nullptr; in FreeInputDeviceMemory()
Dlaunch_mul.h31 input2_addr_(nullptr), in LaunchMul()
52 uint8_t *input2_addr_; variable
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_launch_mul.cc54 …if (!GPUDeviceManager::GetInstance().CopyHostMemToDeviceAsync(input2_addr_, &input2_value_, origin… in CopyHostMemToDevice()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_mul.cc55 …auto ret = rtMemcpyAsync(input2_addr_, dst_size, &input2_value_, origin_size, RT_MEMCPY_HOST_TO_DE… in CopyHostMemToDevice()