Home
last modified time | relevance | path

Searched defs:graph_id (Results 1 – 25 of 48) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/utils/
Dcallbacks_ge.cc70 static TensorPtr GetMeTensorTransformed(uint32_t graph_id, const std::string &parameter_name, in GetMeTensorTransformed()
87 uint32_t CheckpointSaveCallback(uint32_t graph_id, const std::map<std::string, ge::Tensor> &params_… in CheckpointSaveCallback()
158 uint32_t MS_EXPORT SummarySaveCallback(uint32_t graph_id, const std::map<std::string, ge::Tensor> &… in SummarySaveCallback()
Dcallbacks.cc37 uint32_t MS_EXPORT SummarySaveCallback(uint32_t graph_id, const std::map<std::string, TensorPtr> &p… in SummarySaveCallback()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_info.h69 void set_graph_id(uint32_t graph_id) { graph_id_ = graph_id; } in set_graph_id()
70 uint32_t graph_id() const { return graph_id_; } in graph_id() function
Dkernel_runtime.h119 virtual void *GetModelStream(uint32_t graph_id) const { return nullptr; } in GetModelStream()
Dkernel_runtime_manager.cc43 void KernelRuntimeManager::ClearGraphResource(uint32_t graph_id) { in ClearGraphResource()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/
Dacl_multi_graph_session.cc81 void MultiGraphAclSession::RunGraph(GraphId graph_id, const std::vector<MSTensor> &inputs, VectorRe… in RunGraph()
98 VectorRef MultiGraphAclSession::ConstructOutputRef(GraphId graph_id, std::deque<MSTensor> *out_tens… in ConstructOutputRef()
/third_party/mindspore/tests/ut/cpp/stub/tasksink/
Dtask_sink_stub.cc27 uint32_t graph_id) { in GenTasks()
/third_party/mindspore/mindspore/ccsrc/vm/
Dsegment_runner.h39 uint32_t graph_id; member
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
De2e_dump.cc276 uint32_t graph_id = graph->graph_id(); in UpdateIterDumpSetup() local
321 uint32_t graph_id = graph->graph_id(); in DumpData() local
341 bool E2eDump::DumpSingleNodeData(const CNodePtr &node, uint32_t graph_id, uint32_t rank_id, const D… in DumpSingleNodeData()
356 uint32_t graph_id = graph->graph_id(); in DumpParametersAndConstData() local
Dcpu_e2e_dump.cc23 void CPUE2eDump::DumpCNodeData(const CNodePtr &node, uint32_t graph_id) { in DumpCNodeData()
153 void CPUE2eDump::DumpParametersAndConst(const session::KernelGraph *graph, uint32_t graph_id) { in DumpParametersAndConst()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc58 auto graph_id = AscendSession::CompileGraphImpl(func_graph); in CompileGraphImpl() local
87 bool AscendInferenceSession::CheckModelInputs(uint32_t graph_id, const std::vector<tensor::TensorPt… in CheckModelInputs()
Dgpu_inference_session.cc58 auto graph_id = GPUSession::CompileGraphImpl(func_graph); in CompileGraphImpl() local
87 bool GpuInferenceSession::CheckModelInputs(uint32_t graph_id, const std::vector<tensor::TensorPtr> … in CheckModelInputs()
Dsession_basic.h136 virtual bool CheckModelInputs(uint32_t graph_id, const std::vector<tensor::TensorPtr> &inputs, in CheckModelInputs()
242 …virtual void BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfNodePtr, size_t> &paramete… in BuildOpsInGraph()
249 void LoadInputs(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inputs_const) { in LoadInputs()
Dexecutor.cc359 void Executor::RunGraph(const SessionPtr &session, const GraphId &graph_id, in RunGraph()
373 void Executor::RunGraphAsync(const SessionPtr &session, const GraphId &graph_id, in RunGraphAsync()
465 void Executor::RunOpsInGraph(const SessionPtr &session, const GraphId &graph_id, in RunOpsInGraph()
Dsession_basic.cc94 …ByRealKernel(const FuncGraphManagerPtr &manager, const AnfNodePtr &node, const uint32_t graph_id) { in IsUsedByRealKernel()
542 auto graph_id = GetGraphIdByNode(out_node); in InitInternalOutputParameter() local
1663 std::vector<tensor::TensorPtr> SessionBasic::GetInputNeedLockTensors(const GraphId &graph_id, in GetInputNeedLockTensors()
1689 void SessionBasic::CreateOutputTensors(const GraphId &graph_id, const std::vector<tensor::TensorPtr… in CreateOutputTensors()
1755 void SessionBasic::GetModelInputsInfo(uint32_t graph_id, std::vector<tensor::TensorPtr> *inputs, in GetModelInputsInfo()
1784 void SessionBasic::GetModelOutputsInfo(uint32_t graph_id, std::vector<tensor::TensorPtr> *outputs, in GetModelOutputsInfo()
2224 void SessionBasic::BuildGraph(GraphId graph_id) { in BuildGraph()
2236 void SessionBasic::RunOpsInGraph(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inp… in RunOpsInGraph()
2242 void SessionBasic::RunGraph(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inputs, … in RunGraph()
2247 void SessionBasic::RunGraphAsync(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inp… in RunGraphAsync()
[all …]
Dcpu_session.cc115 auto graph_id = graph_sum_; in CompileGraphImpl() local
158 void CPUSession::CreateOutputTensors(const GraphId &graph_id, const std::vector<tensor::TensorPtr> … in CreateOutputTensors()
/third_party/mindspore/mindspore/ccsrc/profiler/device/ascend/
Dmemory_profiling.proto27 int64 graph_id = 1; // graph id field
Dmemory_profiling.cc54 std::shared_ptr<GraphMemory> MemoryProfiling::AddGraphMemoryNode(uint32_t graph_id) { in AddGraphMemoryNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.h42 const std::string &graph_id() const { return graph_id_; } in graph_id() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/
Dprofiling_utils.cc273 void ProfilingUtils::SaveProfilingPoint(uint32_t graph_id, const std::string &node_name, uint32_t p… in SaveProfilingPoint()
370 void ProfilingUtils::SetGraphKernelName(uint32_t graph_id, const std::vector<std::string> &kernel_n… in SetGraphKernelName()
377 void ProfilingUtils::SetGraphProfilingCNode(uint32_t graph_id, const std::vector<CNodePtr> &profili… in SetGraphProfilingCNode()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.cc35 uint32_t graph_id) { in GenTasks()
255 std::vector<TaskInfoPtr> *task_info_list, uint32_t graph_id) { in LaunchAllKernel()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc298 void GPUKernelRuntime::ClearGraphRuntimeResource(uint32_t graph_id) { in ClearGraphRuntimeResource()
466 auto graph_id = graph.graph_id(); in Run() local
505 auto graph_id = graph->graph_id(); in RunOneStep() local
600 auto graph_id = graph->graph_id(); in InitKernelRefCount() local
610 auto graph_id = graph->graph_id(); in InitMemorySwapInfo() local
650 auto graph_id = graph->graph_id(); in SaveGraphOutputNode() local
659 auto graph_id = graph->graph_id(); in IsGraphOutput() local
/third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/
Dgpu_profiling_utils.cc182 bool ProfilingUtils::IsFirstStep(const uint32_t graph_id) { in IsFirstStep()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc506 void GraphCompiler::AddGradAddrToBucket(const GraphId &graph_id, const std::vector<tensor::TensorPt… in AddGradAddrToBucket()
511 void GraphCompiler::ClearAllBucket(const GraphId &graph_id) { in ClearAllBucket()
540 void GraphCompiler::EraseSingleOpCache(const GraphInfo &graph_info, const GraphId &graph_id) { in EraseSingleOpCache()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Ddf_graph_manager.cc138 AnfGraphPtr DfGraphManager::GetAnfGraph(uint32_t graph_id) { in GetAnfGraph()

12