| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 237 AnfAlgo::SetOutputAddr(device_address, i, node.get()); in RunOpMallocPre()
258 AnfAlgo::SetOutputAddr(output_address, index, item.get()); in RunOpMallocPre()
268 AnfAlgo::SetOutputAddr(device_address, index, item.get()); in RunOpMallocPre()
297 AnfAlgo::SetOutputAddr(device_address, index, input_node.get()); in ResetNodeAddress()
304 …AnfAlgo::SetOutputAddr(CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type), … in ResetNodeAddress()
339 AnfAlgo::SetOutputAddr(nullptr, 0, input_node.get()); in RunOpClearMemory()
344 AnfAlgo::SetOutputAddr(nullptr, 0, value_node.get()); in RunOpClearMemory()
351 AnfAlgo::SetOutputAddr(nullptr, index, cnode.get()); in RunOpClearMemory()
415 AnfAlgo::SetOutputAddr(output_address, index, item.get()); in RunOpAssignInputMemory()
431 AnfAlgo::SetOutputAddr(device_address, index, item.get()); in RunOpAssignInputMemory()
[all …]
|
| D | kernel_info.h | 55 bool SetOutputAddr(const DeviceAddressPtr &output_address, size_t index);
|
| D | kernel_info.cc | 48 bool KernelInfo::SetOutputAddr(const DeviceAddressPtr &output_address, size_t index) { in SetOutputAddr() function in mindspore::device::KernelInfo
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 101 AnfAlgo::SetOutputAddr(device_address, index, item.get()); in CreateParameterDeviceAddress()
126 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()),… in CreateDeviceAddressForTensorValue()
142 AnfAlgo::SetOutputAddr(address, output_idx++, value_node.get()); in CreateDeviceAddressForTensorValue()
165 AnfAlgo::SetOutputAddr(address, 0, value_node.get()); in CreateValueNodeDeviceAddress()
190 AnfAlgo::SetOutputAddr(device_address, i, kernel.get()); in CreateKernelOutputDeviceAddress()
250 AnfAlgo::SetOutputAddr(device_address, index, group_node.get()); in UpdateDeviceAddressForInplaceNode()
|
| D | control_node_parser.cc | 171 AnfAlgo::SetOutputAddr(address, 0, front_node.get()); in CreateDeviceTensorForValueNode()
195 AnfAlgo::SetOutputAddr(address, 0, node.get()); in CreateDeviceTensorForFrontParameter()
|
| D | graph_scheduler.cc | 123 AnfAlgo::SetOutputAddr(nullptr, index, input_node.get()); in ClearNodeInfo()
133 AnfAlgo::SetOutputAddr(nullptr, 0, value_node.get()); in ClearNodeInfo()
142 AnfAlgo::SetOutputAddr(nullptr, index, cnode.get()); in ClearNodeInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | data_prepare_actor.cc | 258 AnfAlgo::SetOutputAddr(tensor_address, 0, input_node.get()); in PrepareDataForHostTensorQueue()
304 AnfAlgo::SetOutputAddr(host_tensor_address, 0, input_node.get()); in PrepareDataForStepMode()
316 AnfAlgo::SetOutputAddr(device_address, 0, input_node.get()); in PrepareDataForStepMode()
425 AnfAlgo::SetOutputAddr(host_tensor_address, 0, backend_node.get()); in PrepareDataForWeightNode()
571 AnfAlgo::SetOutputAddr(tensor_address, 0, backend_node.get()); in PrepareHostTensorQueueForControlNode()
|
| D | output_actor.cc | 100 AnfAlgo::SetOutputAddr(new_device_tensor, output_index, output_node.get()); in UpdateOutputDeviceAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 102 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()),… in AssignValueNodeAddress()
126 AnfAlgo::SetOutputAddr(address, 0, item_node.get()); in AssignValueNodeAddress()
149 AnfAlgo::SetOutputAddr(address, index, item.get()); in AssignInputNodeAddress()
165 …AnfAlgo::SetOutputAddr(CreateDeviceAddress(nullptr, output_sizes[i], output_format, output_type), … in AssignKernelOutputAddress()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 518 d_kernel_info->SetOutputAddr(device_address, 0); in TEST_F()
533 d_kernel_info->SetOutputAddr(device_address, 0); in TEST_F()
543 TEST_F(AnfRuntimeAlgorithmTest, SetOutputAddr) { in TEST_F() argument
550 EXPECT_THROW(AnfAlgo::SetOutputAddr(device_address, 0, nullptr), std::runtime_error); in TEST_F()
551 AnfAlgo::SetOutputAddr(device_address, 0, add.get()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.cc | 354 AnfAlgo::SetOutputAddr(tensor_address, 0, pk_node.get()); in LoadInputData()
603 AnfAlgo::SetOutputAddr(new_address, output_index, node.get()); in UpdateOutputTensors()
|
| D | anf_runtime_algorithm.h | 200 static void SetOutputAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode *node);
|
| D | anf_runtime_algorithm.cc | 1116 void AnfRuntimeAlgorithm::SetOutputAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode *n… in SetOutputAddr() function in mindspore::session::AnfRuntimeAlgorithm
1120 if (!kernel_info->SetOutputAddr(addr, output_idx)) { in SetOutputAddr()
2362 SetOutputAddr(device_address, idx, kernel.get()); in CacheAddrForGraph()
|
| D | session_basic.cc | 324 AnfAlgo::SetOutputAddr(device_address, 0, param.get()); in ConstructRunOpParameter()
583 AnfAlgo::SetOutputAddr(address, 0, parameter.get()); in InitInternalOutputParameter()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 257 AnfAlgo::SetOutputAddr(new_device_tensor, output_index, output_node.get()); in UpdateOutputDeviceAddress()
267 AnfAlgo::SetOutputAddr(nullptr, 0, node.get()); in UpdateInputDeviceAddress()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 353 AnfAlgo::SetOutputAddr(device_address, index, node.get()); in AllocInplaceNodeMemory()
629 AnfAlgo::SetOutputAddr(device_address, i, kernel.get()); in InitKernelOutputAddress()
|