Home
last modified time | relevance | path

Searched defs:GetOutputAddr (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pipeline/task/
Dkernel_task.h52 const device::DeviceAddressPtr GetOutputAddr(size_t idx) { in GetOutputAddr() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dkernel_info.cc62 const DeviceAddress *KernelInfo::GetOutputAddr(size_t index) const { in GetOutputAddr() function in mindspore::device::KernelInfo
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc522 TEST_F(AnfRuntimeAlgorithmTest, GetOutputAddr) { in TEST_F() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc820 const DeviceAddress *AnfRuntimeAlgorithm::GetOutputAddr(const AnfNodePtr &node, size_t output_idx, … in GetOutputAddr() function in mindspore::session::AnfRuntimeAlgorithm