Home
last modified time | relevance | path

Searched refs:outputs_addr_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.cc61 outputs_addr_ = std::vector<uint8_t *>(outputs_list.size(), nullptr); in ObtainKernelOutputs()
62 auto kernel_outputs = ObtainKernelAddress(outputs_list, &outputs_addr_); in ObtainKernelOutputs()
94 for (size_t i = 0; i < outputs_addr_.size(); ++i) { in FreeOutputAndWorkspaceDeviceMem()
95 if (outputs_addr_[i] != nullptr) { in FreeOutputAndWorkspaceDeviceMem()
96 FreeDeviceMem(outputs_addr_[i]); in FreeOutputAndWorkspaceDeviceMem()
97 outputs_addr_[i] = nullptr; in FreeOutputAndWorkspaceDeviceMem()
106 outputs_addr_.clear(); in FreeOutputAndWorkspaceDeviceMem()
Dlaunch_kernel.h30 std::vector<uint8_t *> GetKernelOutputAddr() { return outputs_addr_; } in GetKernelOutputAddr()
47 std::vector<uint8_t *> outputs_addr_; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel.h194 void set_outputs_addr(const std::vector<AddressPtr> &addr) { outputs_addr_ = addr; } in set_outputs_addr()
197 const std::vector<AddressPtr> &GetOutputsAddr() { return outputs_addr_; } in GetOutputsAddr()
211 std::vector<AddressPtr> outputs_addr_; variable