Searched refs:SetWorkspaceAddr (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.h | 59 bool SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index);
|
| D | kernel_info.cc | 91 bool KernelInfo::SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index) { in SetWorkspaceAddr() function in mindspore::device::KernelInfo
|
| D | kernel_runtime.cc | 309 …AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(nullptr, workspace_sizes[i], kOpFormat_DEFAULT, kNum… in ResetNodeAddress()
358 AnfAlgo::SetWorkspaceAddr(nullptr, index, cnode.get()); in RunOpClearMemory()
498 AnfAlgo::SetWorkspaceAddr(device_address, i, kernel.get()); in RunOpAssignWorkSpaceMemory()
1115 AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(ptr, size, "", kTypeUnknown), index, node.get()); in AssignWorkSpaceMem()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 566 d_kernel_info->SetWorkspaceAddr(device_address, 0); in TEST_F()
570 TEST_F(AnfRuntimeAlgorithmTest, SetWorkspaceAddr) { in TEST_F() argument
577 EXPECT_THROW(AnfAlgo::SetWorkspaceAddr(device_address, 0, nullptr), std::runtime_error); in TEST_F()
578 AnfAlgo::SetWorkspaceAddr(device_address, 0, add.get()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 202 static void SetWorkspaceAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode *node);
|
| D | anf_runtime_algorithm.cc | 1127 void AnfRuntimeAlgorithm::SetWorkspaceAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode… in SetWorkspaceAddr() function in mindspore::session::AnfRuntimeAlgorithm
1131 if (!kernel_info->SetWorkspaceAddr(addr, output_idx)) { in SetWorkspaceAddr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 209 AnfAlgo::SetWorkspaceAddr(device_address, i, kernel.get()); in CreateKernelWorkspaceDeviceAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 170 …AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(nullptr, workspace_sizes[i], kOpFormat_DEFAULT, kNum… in AssignKernelOutputAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 643 AnfAlgo::SetWorkspaceAddr(device_address, i, kernel.get()); in InitKernelWorkspaceAddress()
|