Searched defs:KernelGraph (Results 1 – 12 of 12) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
D | gpu_session.h | 27 using KernelGraph = mindspore::session::KernelGraph; variable
|
D | kernel_graph.h | 56 …KernelGraph() : 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/ |
D | matmul_confusiontranspose_fusion_pass.cc | 27 … const session::KernelGraph & /* kernel_graph */, in MatchMatmulConfusionTranpose()
|
D | conv2dbackprop_eltwise_fusion_pass.cc | 26 … const session::KernelGraph & /* kernel_graph */, in MatchConv2DBackpropInputEltwise()
|
D | batchmatmul_fusedmuladd_fusion_pass.cc | 26 … const session::KernelGraph & /* kernel_graph */, in MatchBatchMatmulFusedMulAdd()
|
D | matmul_eltwise_fusion_pass.cc | 26 const session::KernelGraph & /* kernel_graph */, in MatchMatmulEltwise()
|
D | depthwiseconv_eltwise_fusion_pass.cc | 27 … const session::KernelGraph & /* kernel_graph */, in MatchDepthwiseConvRelu()
|
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
D | mem_reuse.h | 39 using KernelGraph = mindspore::session::KernelGraph; variable
|
/third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
D | debugger_utils.cc | 32 using KernelGraph = mindspore::session::KernelGraph; typedef
|
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
D | gpu_device_context.cc | 52 using KernelGraph = mindspore::session::KernelGraph; typedef
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
D | ascend_kernel_runtime.cc | 373 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/ |
D | kernel_runtime.cc | 78 bool KernelRuntime::Load(const session::KernelGraph &, bool) { in Load() 83 bool KernelRuntime::LoadData(const session::KernelGraph &) { in LoadData()
|