| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | cpu_e2e_dump.cc | 70 ShapeVector int_shapes; in DumpInputImpl() local
71 GetDumpIntShape(input, index, NOT_NULL(&int_shapes)); in DumpInputImpl()
81 DumpMemToFile(file_path, *addr, int_shapes, type); in DumpInputImpl()
95 ShapeVector int_shapes; in DumpOutputImpl() local
96 GetDumpIntShape(node, j, NOT_NULL(&int_shapes)); in DumpOutputImpl()
106 DumpMemToFile(file_path, *addr, int_shapes, type); in DumpOutputImpl()
141 ShapeVector int_shapes; in DumpSingleAnfNode() local
142 GetDumpIntShape(anf_node, output_index, NOT_NULL(&int_shapes)); in DumpSingleAnfNode()
150 DumpMemToFile(file_path, *addr, int_shapes, type); in DumpSingleAnfNode()
|
| D | e2e_dump.cc | 46 const device::DeviceAddress &addr, const ShapeVector &int_shapes, in DumpGPUMemToFile() argument
52 …ugger->DumpTensorToFile(original_kernel_name, trans_flag, file_path, format, int_shapes, host_type, in DumpGPUMemToFile()
107 ShapeVector int_shapes; in DumpOutputImpl() local
108 GetDumpIntShape(node, j, NOT_NULL(&int_shapes), trans_flag); in DumpOutputImpl()
120 …DumpGPUMemToFile(file_path, GetKernelNodeName(node), *addr, int_shapes, type, device_type, trans_f… in DumpOutputImpl()
123 DumpMemToFile(file_path, *addr, int_shapes, type, trans_flag); in DumpOutputImpl()
186 ShapeVector int_shapes; in DumpInputImpl() local
187 GetDumpIntShape(input, index, NOT_NULL(&int_shapes), trans_flag); in DumpInputImpl()
199 …DumpGPUMemToFile(file_path, tensor_name, *addr, int_shapes, type, device_type, trans_flag, slot, d… in DumpInputImpl()
201 DumpMemToFile(file_path, *addr, int_shapes, type, trans_flag); in DumpInputImpl()
[all …]
|
| D | dump_utils.cc | 114 void GetDumpIntShape(const AnfNodePtr &node, size_t index, NotNull<ShapeVector *> int_shapes, bool … in GetDumpIntShape() argument
116 *int_shapes = trans::GetRuntimePaddingShape(node, index); in GetDumpIntShape()
119 (void)std::transform(shape.begin(), shape.end(), std::back_inserter(*int_shapes), in GetDumpIntShape()
124 …ile(const std::string &file_path, const device::DeviceAddress &addr, const ShapeVector &int_shapes, in DumpMemToFile() argument
127 auto ret = addr.DumpMemToFile(file_path, format, int_shapes, type, trans_flag); in DumpMemToFile()
|
| D | dump_utils.h | 36 void GetDumpIntShape(const AnfNodePtr &node, size_t index, NotNull<ShapeVector *> int_shapes, bool …
38 …ile(const std::string &file_path, const device::DeviceAddress &addr, const ShapeVector &int_shapes,
|
| D | e2e_dump.h | 73 const device::DeviceAddress &addr, const ShapeVector &int_shapes,
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
| D | debugger_utils.cc | 72 ShapeVector int_shapes = trans::GetRuntimePaddingShape(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadInputs() local
73 …auto ret = gpu_addr->LoadMemToHost(input_tensor_name, exec_order_, format, int_shapes, type, 0, tr… in LoadInputs()
101 ShapeVector int_shapes = trans::GetRuntimePaddingShape(cnode, j); in LoadOutputs() local
102 … auto ret = gpu_addr->LoadMemToHost(tensor_name, exec_order_, format, int_shapes, type, j, false); in LoadOutputs()
|
| D | debugger.cc | 1342 ShapeVector int_shapes = trans::GetRuntimePaddingShape(anf_node, output_index); in LoadSingleAnfnode() local
1350 bool ret = addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, 0, keep_prev); in LoadSingleAnfnode()
1422 ShapeVector int_shapes = trans::GetRuntimePaddingShape(node, j); in LoadGraphOutputs() local
1423 auto ret = addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, j, false); in LoadGraphOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 176 ShapeVector int_shapes = trans::GetRuntimePaddingShape(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadKernelData() local
177 …auto ret = gpu_addr->LoadMemToHost(input_tensor_name, exec_order, format, int_shapes, type, 0, tru… in LoadKernelData()
203 ShapeVector int_shapes = trans::GetRuntimePaddingShape(kernel, j); in LoadKernelData() local
204 … auto ret = gpu_addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, j, false); in LoadKernelData()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 1188 std::vector<ShapeVector> int_shapes; in InitExecDatasetVm() local
1189 (void)std::transform(shapes.begin(), shapes.end(), std::back_inserter(int_shapes), in InitExecDatasetVm()
1201 p_init->set_attr("shapes", MakeValue(int_shapes)); in InitExecDatasetVm()
|