Home
last modified time | relevance | path

Searched defs:kernel_graph (Results 1 – 25 of 112) sorted by relevance

12345

/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc40 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
99 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
110 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
122 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
131 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
145 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
169 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
193 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
208 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
221 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
[all …]
Dkernel_graph_test.cc38 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
61 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
98 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
109 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
140 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
187 auto kernel_graph = std::make_shared<KernelGraph>(); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dfusion_base_pass.cc23 bool FusionBasePass::CheckEltWiseNode(const session::KernelGraph &kernel_graph, const AnfNodePtr &n… in CheckEltWiseNode()
38 bool FusionBasePass::CheckDoubleInEltWiseNode(const session::KernelGraph &kernel_graph, const AnfNo… in CheckDoubleInEltWiseNode()
53 bool FusionBasePass::CheckMultiOutputEltWiseNode(const session::KernelGraph &kernel_graph, const An… in CheckMultiOutputEltWiseNode()
68 size_t FusionBasePass::GetNotUpdateStateUserNums(const session::KernelGraph &kernel_graph, const An… in GetNotUpdateStateUserNums()
90 bool FusionBasePass::MatchUBFusionPattern(const session::KernelGraph &kernel_graph) { in MatchUBFusionPattern()
110 auto kernel_graph = graph->cast<std::shared_ptr<session::KernelGraph>>(); in Run() local
Dub_pattern_fusion.cc64 … const std::vector<AnfNodePtr> &anf_nodes, session::KernelGraph *kernel_graph) { in CreateFusionOp()
144 …ePtr CreateTupleGetItem(const AnfNodePtr &buffer_fusion_kernel, session::KernelGraph *kernel_graph, in CreateTupleGetItem()
181 const AnfNodePtr &buffer_fusion_kernel, session::KernelGraph *kernel_graph) { in ReplaceOldNode()
207 void GetFusionScopeComputeNodeList(session::KernelGraph *kernel_graph, in GetFusionScopeComputeNodeList()
225 void GetFusionScopeInputNodeList(const session::KernelGraph &kernel_graph, in GetFusionScopeInputNodeList()
270 AnfNodePtr RemoveNodeFromUpdateState(session::KernelGraph *kernel_graph, const AnfNodePtr &node, in RemoveNodeFromUpdateState()
286 void GetFusionScopeOutputNodeList(session::KernelGraph *kernel_graph, in GetFusionScopeOutputNodeList()
352 void SetOutputUsedNumAttr(const session::KernelGraph &kernel_graph, in SetOutputUsedNumAttr()
364 void SetFusionOpRefInfos(session::KernelGraph *kernel_graph, const std::vector<AnfNodePtr> &outputs… in SetFusionOpRefInfos()
395 bool CheckCircle(const session::KernelGraph &kernel_graph, const BufferFusionInfo_t &fusion_info) { in CheckCircle()
[all …]
Dconv_bnreduce_fusion_pass.cc27 …educeFusionPass::MatchConvBnreduce(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchConvBnreduce()
41 void ConvBnReduceFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Deltwise_fusion_pass.cc25 …id EltwiseFusionPass::MatchEltwise(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchEltwise()
51 void EltwiseFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dconv_single_in_fusion_pass.cc29 …sionPass::MatchConvSingleInEltwise(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchConvSingleInEltwise()
57 void ConvSingleInFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dconv_double_in_fusion_pass.cc25 …sionPass::MatchConvDoubleInEltwise(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchConvDoubleInEltwise()
53 void ConvDoubleInFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dmulti_output_fusion_pass.cc26 …usionPass::MatchMultiOutputEltwise(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchMultiOutputEltwise()
57 void MultiOutputFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dbnupdate_eltwise_fusion_pass.cc27 … const session::KernelGraph &kernel_graph, in MatchBnupdateDoubleOutputEltwise()
44 void BnupdateEltwiseFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dstridedread_conv_stridedwrite_fusion_pass.cc26 … const session::KernelGraph &kernel_graph, in MatchStridedReadConvStridedWrite()
64 …dReadConvStridedWriteFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dconv2dbackprop_eltwise_eltwise_fusion_pass.cc26 …const CNodePtr &cnode, const session::KernelGraph &kernel_graph, FusedNodeRecord *candidate_fusion… in MatchConv2DBackpropInputEltwiseEltwise()
66 …ackpropEltwiseEltwiseFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dreduce_eltwise_fusion_pass.cc27 …wiseFusionPass::MatchReduceEltwise(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchReduceEltwise()
69 void ReduceEltwiseFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dbnupdate_eltwise_eltwise_fusion_pass.cc32 … const session::KernelGraph &kernel_graph, in MatchBnupdateAddRelu()
55 …nupdateEltwiseEltwiseFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
Dsegment_eltwise_fusion_pass.cc25 …iseFusionPass::MatchSegmentEltwise(const CNodePtr &cnode, const session::KernelGraph &kernel_graph, in MatchSegmentEltwise()
66 void SegmentEltwiseFusionPass::MatchSingleFusionPattern(const session::KernelGraph &kernel_graph, in MatchSingleFusionPattern()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_backend_optimization.cc222 void AscendDataLayout(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AscendDataLayout()
255 void AscendMixPrecision(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AscendMixPrecision()
280 void AscendBackendIRFusionOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AscendBackendIRFusionOptimization()
325 void RunOpAscendBackendIRFusionOptimization(const std::shared_ptr<session::KernelGraph> &kernel_gra… in RunOpAscendBackendIRFusionOptimization()
377 void RunOpAscendBackendOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in RunOpAscendBackendOptimization()
391 void AscendBackendOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AscendBackendOptimization()
455 void AscendBackendUBFusionOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AscendBackendUBFusionOptimization()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc81 void CPUSession::Optimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in Optimize()
102 void CPUSession::ProcessCast(const std::shared_ptr<KernelGraph> &kernel_graph) { in ProcessCast()
161 auto kernel_graph = GetGraph(graph_id); in CreateOutputTensors() local
166 void CPUSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, in LoadInputData()
199 void CPUSession::PreExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PreExecuteGraph()
209 void CPUSession::PostExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PostExecuteGraph()
216 void CPUSession::ExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph) { in ExecuteGraph()
233 const auto &kernel_graph = ConstructSingleOpGraph(op_run_info, input_tensors, tensors_mask); in BuildOpImpl() local
284 const auto &kernel_graph = BuildOpImpl(*op_run_info, graph_info, *input_tensors, tensors_mask); in RunOpImpl() local
313 void CPUSession::SetKernelInfo(const KernelGraph *kernel_graph) { in SetKernelInfo()
[all …]
Dgpu_session.cc153 void GPUSession::Optimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in Optimize()
180 void GPUSession::HardwareOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in HardwareOptimize()
208 void GPUSession::RunOpOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in RunOpOptimize()
219 void GPUSession::RunOpHardwareOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in RunOpHardwareOptimize()
229 void GPUSession::GraphKernelOptimize(const std::shared_ptr<KernelGraph> &kernel_graph) { in GraphKernelOptimize()
237 void GPUSession::AssignStream(const std::shared_ptr<KernelGraph> &kernel_graph) { in AssignStream()
279 bool UpdatedByAssign(const KernelGraphPtr &kernel_graph, const AnfNodePtr &node) { in UpdatedByAssign()
319 void GPUSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, in LoadInputData()
507 void GPUSession::PreExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PreExecuteGraph()
523 void GPUSession::PostExecuteGraph(const std::shared_ptr<KernelGraph> &kernel_graph, in PostExecuteGraph()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dcommon_backend_optimization.cc35 void BackendCommonOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in BackendCommonOptimization()
68 void CommonFinalOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in CommonFinalOptimization()
89 void CommonUnifyMindIROptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in CommonUnifyMindIROptimization()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_stream_assign.cc30 void AssignGpuStream(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in AssignGpuStream()
62 bool FindAllReduceStreamSwitchPos(const std::shared_ptr<session::KernelGraph> &kernel_graph, in FindAllReduceStreamSwitchPos()
146 void InsertStreamSwitchNode(const std::shared_ptr<session::KernelGraph> &kernel_graph, in InsertStreamSwitchNode()
178 bool GenSendRecvCNodesForAllReduce(const std::shared_ptr<session::KernelGraph> &kernel_graph, in GenSendRecvCNodesForAllReduce()
200 CNodePtr CreateStreamSwitchNode(const std::shared_ptr<session::KernelGraph> &kernel_graph, const st… in CreateStreamSwitchNode()
218 void CacheSendRecvCNodesForAllReduce(const std::shared_ptr<session::KernelGraph> &kernel_graph, in CacheSendRecvCNodesForAllReduce()
Dgpu_launch_mul.cc31 void GPULaunchMul::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelSelect()
35 void GPULaunchMul::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc62 void CPUKernelRuntime::AssignKernelAddress(session::KernelGraph *kernel_graph) { in AssignKernelAddress()
87 void CPUKernelRuntime::AssignValueNodeAddress(session::KernelGraph *kernel_graph) { in AssignValueNodeAddress()
131 void CPUKernelRuntime::AssignInputNodeAddress(const session::KernelGraph *kernel_graph) { in AssignInputNodeAddress()
155 void CPUKernelRuntime::AssignKernelOutputAddress(const session::KernelGraph *kernel_graph) { in AssignKernelOutputAddress()
187 session::KernelGraph *kernel_graph, const CNodePtr &node, size_t index, in CreatTensorForOutput()
247 BaseRef CPUKernelRuntime::CreatTensorForOutput(session::KernelGraph *kernel_graph, in CreatTensorForOutput()
279 void CPUKernelRuntime::CreateOutputTensors(session::KernelGraph *kernel_graph, in CreateOutputTensors()
307 void CPUKernelRuntime::BindInputTensorAddressPtr(const session::KernelGraph &kernel_graph, in BindInputTensorAddressPtr()
383 void CPUKernelRuntime::BindInputOutput(session::KernelGraph *kernel_graph, const std::vector<tensor… in BindInputOutput()
413 bool CPUKernelRuntime::Run(const session::KernelGraph &kernel_graph, bool) { in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/
Dprofiling_utils.cc55 ProfilingTraceInfo ProfilingUtils::GenerateProfilingTrace(const session::KernelGraph &kernel_graph)… in GenerateProfilingTrace()
84 void ProfilingUtils::GetTraceHccl(const session::KernelGraph &kernel_graph, in GetTraceHccl()
95 void ProfilingUtils::GetTraceBegin(const session::KernelGraph &kernel_graph, const nlohmann::json &… in GetTraceBegin()
129 …ils::GetCNodeOutputRealNode(const std::string &node_name, const session::KernelGraph &kernel_graph, in GetCNodeOutputRealNode()
151 void ProfilingUtils::GetTraceBpEnd(const session::KernelGraph &kernel_graph, const nlohmann::json &… in GetTraceBpEnd()
203 std::string ProfilingUtils::GetGraphLastKernelName(const session::KernelGraph &kernel_graph) { in GetGraphLastKernelName()
221 void ProfilingUtils::GetTraceIterEnd(const session::KernelGraph &kernel_graph, ProfilingTraceInfo *… in GetTraceIterEnd()
384 bool ProfilingUtils::ValidComputeGraph(const session::KernelGraph &kernel_graph) { in ValidComputeGraph()
394 const session::KernelGraph &kernel_graph) { in ReportProfilingData()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_kernel.cc34 void AscendLaunchKernel::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelSelect()
45 void AscendLaunchKernel::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild()
Dascend_launch_mul.cc32 void AscendLaunchMul::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelSelect()
36 void AscendLaunchMul::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild()

12345