Home
last modified time | relevance | path

Searched refs:GetWorkspaceAddr (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_simple_mem_plan.cc54 auto address = AnfAlgo::GetWorkspaceAddr(kernel, i); in MemPlan()
100 auto address = AnfAlgo::GetWorkspaceAddr(kernel, i); in MemAssign()
Dcpu_memory_manager.cc185 auto address = AnfAlgo::GetWorkspaceAddr(kernel, i); in IncreaseAddressRefCount()
210 auto address = AnfAlgo::GetWorkspaceAddr(kernel, i); in DecreaseAddressRefCount()
Dcpu_kernel_runtime.cc452 auto device_address = AnfAlgo::GetWorkspaceAddr(kernel, i); in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_info.h56 DeviceAddress *GetWorkspaceAddr(size_t index) const;
Dkernel_info.cc68 DeviceAddress *KernelInfo::GetWorkspaceAddr(size_t index) const { in GetWorkspaceAddr() function in mindspore::device::KernelInfo
Dkernel_runtime.cc1171 auto device_address = AnfAlgo::GetWorkspaceAddr(kernel, i); in GenLaunchArgs()
1220 auto device_address = AnfAlgo::GetWorkspaceAddr(pre_node, index); in GenAddrCleanLaunchArgs()
1335 auto device_address = AnfAlgo::GetWorkspaceAddr(kernel, i); in AssignKernelAddress()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc126 auto device_address = AnfAlgo::GetWorkspaceAddr(pre_node, index); in LaunchAddrCleanKernel()
221 auto device_address = AnfAlgo::GetWorkspaceAddr(anf_node_ptr, i); in LaunchKernel()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc555 TEST_F(AnfRuntimeAlgorithmTest, GetWorkspaceAddr) { in TEST_F() argument
567 EXPECT_EQ(AnfAlgo::GetWorkspaceAddr(add, 0), device_address.get()); in TEST_F()
579 EXPECT_EQ(AnfAlgo::GetWorkspaceAddr(add, 0), device_address.get()); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.h204 static DeviceAddress *GetWorkspaceAddr(const AnfNodePtr &node, size_t output_idx);
Danf_runtime_algorithm.cc1138 DeviceAddress *AnfRuntimeAlgorithm::GetWorkspaceAddr(const AnfNodePtr &node, size_t output_idx) { in GetWorkspaceAddr() function in mindspore::session::AnfRuntimeAlgorithm
1142 auto addr = kernel_info->GetWorkspaceAddr(output_idx); in GetWorkspaceAddr()
2421 auto device_address = GetWorkspaceAddr(node, i); in CacheAddrForKernel()
2471 auto device_address = GetWorkspaceAddr(pre_node, index); in CacheAddrForAtomicClean()