Home
last modified time | relevance | path

Searched defs:KernelGraph (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.h27 using KernelGraph = mindspore::session::KernelGraph; variable
Dkernel_graph.h56KernelGraph() : graph_id_(0), start_label_(nullptr), end_goto_(nullptr), current_epoch_(0), is_dyn… in KernelGraph() function
65 KernelGraph(const KernelGraph &graph) : FuncGraph(graph) { in KernelGraph() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dmatmul_confusiontranspose_fusion_pass.cc27const session::KernelGraph & /* kernel_graph */, in MatchMatmulConfusionTranpose()
Dconv2dbackprop_eltwise_fusion_pass.cc26const session::KernelGraph & /* kernel_graph */, in MatchConv2DBackpropInputEltwise()
Dbatchmatmul_fusedmuladd_fusion_pass.cc26const session::KernelGraph & /* kernel_graph */, in MatchBatchMatmulFusedMulAdd()
Dmatmul_eltwise_fusion_pass.cc26 const session::KernelGraph & /* kernel_graph */, in MatchMatmulEltwise()
Ddepthwiseconv_eltwise_fusion_pass.cc27const session::KernelGraph & /* kernel_graph */, in MatchDepthwiseConvRelu()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.h39 using KernelGraph = mindspore::session::KernelGraph; variable
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger_utils.cc32 using KernelGraph = mindspore::session::KernelGraph; typedef
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc52 using KernelGraph = mindspore::session::KernelGraph; typedef
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc373 bool AscendKernelRuntime::LoadData(const session::KernelGraph & /* graph */) { in LoadData()
676 void AscendKernelRuntime::DumpTaskExceptionInfo(const session::KernelGraph & /* graph */) { in DumpTaskExceptionInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc78 bool KernelRuntime::Load(const session::KernelGraph &, bool) { in Load()
83 bool KernelRuntime::LoadData(const session::KernelGraph &) { in LoadData()