Home
last modified time | relevance | path

Searched defs:int_shapes (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Dcpu_e2e_dump.cc70 ShapeVector int_shapes; in DumpInputImpl() local
95 ShapeVector int_shapes; in DumpOutputImpl() local
141 ShapeVector int_shapes; in DumpSingleAnfNode() local
De2e_dump.cc46 const device::DeviceAddress &addr, const ShapeVector &int_shapes, in DumpGPUMemToFile()
107 ShapeVector int_shapes; in DumpOutputImpl() local
186 ShapeVector int_shapes; in DumpInputImpl() local
235 ShapeVector int_shapes; in DumpSingleAnfNode() local
Ddump_utils.cc114 void GetDumpIntShape(const AnfNodePtr &node, size_t index, NotNull<ShapeVector *> int_shapes, bool … in GetDumpIntShape()
124 …ile(const std::string &file_path, const device::DeviceAddress &addr, const ShapeVector &int_shapes, in DumpMemToFile()
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger_utils.cc72 ShapeVector int_shapes = trans::GetRuntimePaddingShape(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadInputs() local
101 ShapeVector int_shapes = trans::GetRuntimePaddingShape(cnode, j); in LoadOutputs() local
Ddebugger.cc1342 ShapeVector int_shapes = trans::GetRuntimePaddingShape(anf_node, output_index); in LoadSingleAnfnode() local
1422 ShapeVector int_shapes = trans::GetRuntimePaddingShape(node, j); in LoadGraphOutputs() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc176 ShapeVector int_shapes = trans::GetRuntimePaddingShape(input_kernel, PARAMETER_OUTPUT_INDEX); in LoadKernelData() local
203 ShapeVector int_shapes = trans::GetRuntimePaddingShape(kernel, j); in LoadKernelData() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc1188 std::vector<ShapeVector> int_shapes; in InitExecDatasetVm() local