| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | cpu_e2e_dump.cc | 69 auto addr = AnfAlgo::GetOutputAddr(input, index); in DumpInputImpl()
93 auto addr = AnfAlgo::GetOutputAddr(node, j); in DumpOutputImpl()
136 auto addr = AnfAlgo::GetOutputAddr(anf_node, output_index); in DumpSingleAnfNode()
|
| D | e2e_dump.cc | 105 auto addr = AnfAlgo::GetOutputAddr(node, j); in DumpOutputImpl()
175 auto addr = AnfAlgo::GetOutputAddr(input, index); in DumpInputImpl()
233 auto addr = AnfAlgo::GetOutputAddr(anf_node, output_index); in DumpSingleAnfNode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_simple_mem_plan.cc | 35 … auto address = AnfAlgo::GetOutputAddr(kernel_with_index.first, kernel_with_index.second, true); in MemPlan()
44 auto address = AnfAlgo::GetOutputAddr(kernel, i); in MemPlan()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_shape_kernel.cc | 44 auto output_addr = AnfAlgo::GetOutputAddr(cnode, 0); in Execute()
|
| D | dynamic_broadcast_gradient_args_kernel.cc | 159 auto out_addr = AnfAlgo::GetOutputAddr(cnode, index); in SetOutputValue()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 84 auto device_address = AnfAlgo::GetOutputAddr(post_node, index); in LaunchAddrCleanAkgKernel()
112 auto device_address = AnfAlgo::GetOutputAddr(pre_node, index); in LaunchAddrCleanKernel()
212 auto it = AnfAlgo::GetOutputAddr(anf_node_ptr, i); in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.h | 52 const DeviceAddress *GetOutputAddr(size_t index) const;
|
| D | kernel_info.cc | 25 const DeviceAddress *KernelInfo::GetOutputAddr(size_t index) const { in GetOutputAddr() function in mindspore::device::KernelInfo
|
| D | kernel_runtime.cc | 91 const auto &address = AnfAlgo::GetOutputAddr(kernel, index); in NodeOutputDeviceAddressExist()
1161 auto device_address = AnfAlgo::GetOutputAddr(kernel, i, visit_nop_node); in GenLaunchArgs()
1198 auto device_address = AnfAlgo::GetOutputAddr(pre_node, index); in GenAddrCleanLaunchArgs()
1318 auto device_address = AnfAlgo::GetOutputAddr(input_node, index, true); in AssignKernelAddress()
1327 auto device_address = AnfAlgo::GetOutputAddr(kernel, j, true); in AssignKernelAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_cpu_dynamic_kernel.cc | 122 auto output_addr = AnfAlgo::GetOutputAddr(cnode, i); in UpdateInputOutputAddr()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 507 TEST_F(AnfRuntimeAlgorithmTest, GetOutputAddr) { in TEST_F() argument
519 EXPECT_EQ(AnfAlgo::GetOutputAddr(add, 0), device_address.get()); in TEST_F()
552 EXPECT_EQ(AnfAlgo::GetOutputAddr(add, 0), device_address.get()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 187 …static const DeviceAddress *GetOutputAddr(const AnfNodePtr &node, size_t output_idx, bool visit_no…
|
| D | anf_runtime_algorithm.cc | 1029 const DeviceAddress *AnfRuntimeAlgorithm::GetOutputAddr(const AnfNodePtr &node, size_t output_idx, in GetOutputAddr() function in mindspore::session::AnfRuntimeAlgorithm
1044 auto addr = kernel_info->GetOutputAddr(output_idx); in GetOutputAddr()
1106 …return AnfRuntimeAlgorithm::GetOutputAddr(kernel_with_index.first, kernel_with_index.second, visit… in GetPrevNodeOutputAddr()
2412 auto device_address = GetOutputAddr(node, i, visit_nop_node); in CacheAddrForKernel()
2453 auto device_address = GetOutputAddr(pre_node, index); in CacheAddrForAtomicClean()
|
| D | ascend_session.cc | 1746 auto addr = AnfAlgo::GetOutputAddr(backend_parameter, 0); in SyncInitialTenosrToDevice()
|
| D | session_basic.cc | 1870 auto address = AnfAlgo::GetOutputAddr(node, index); in Summary()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 430 auto address = AnfAlgo::GetOutputAddr(kernel, i); in DumpKernelOutput()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger.cc | 1334 auto addr = AnfAlgo::GetOutputAddr(anf_node, output_index); in LoadSingleAnfnode()
1414 auto addr = AnfAlgo::GetOutputAddr(node, j); in LoadGraphOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse_checker.cc | 366 auto it = AnfAlgo::GetOutputAddr(node, i); in CheckNormalIR()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 50 const auto &address = AnfAlgo::GetOutputAddr(kernel, index); in NodeDeviceAddressExist()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 639 auto device_addr = AnfAlgo::GetOutputAddr(node, index); in CreateSomasParameter()
|