Home
last modified time | relevance | path

Searched defs:node_index (Results 1 – 25 of 206) sorted by relevance

123456789

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Drelux_int8_coder.h32 const LiteGraph::Node *node, size_t node_index, Target target) in ReluxInt8Coder()
51 const LiteGraph::Node *node, size_t node_index, Target target) in ReluInt8Coder()
67 const LiteGraph::Node *node, size_t node_index, Target target) in Relu6Int8Coder()
Dleaky_relu_int8_coder.h31 const LiteGraph::Node *node, size_t node_index, Target target) in LeakyReluInt8Coder()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dgru_fusion_pass.cc53 bool CheckCommon(schema::MetaGraphT *graph, uint32_t node_index, schema::PrimitiveType type, size_t… in CheckCommon()
79 bool CheckArithmetic(schema::MetaGraphT *graph, uint32_t node_index) { in CheckArithmetic()
105 bool CheckActivation(schema::MetaGraphT *graph, uint32_t node_index) { in CheckActivation()
117 bool CheckBiasAdd(schema::MetaGraphT *graph, uint32_t node_index) { in CheckBiasAdd()
138 bool CheckMatmul(schema::MetaGraphT *graph, uint32_t node_index) { in CheckMatmul()
152 bool CheckSplit(schema::MetaGraphT *graph, uint32_t node_index) { in CheckSplit()
176 bool CheckStack(schema::MetaGraphT *graph, uint32_t node_index) { in CheckStack()
194 bool CheckSqueeze(schema::MetaGraphT *graph, uint32_t node_index) { in CheckSqueeze()
242 bool CheckStridedSlice(schema::MetaGraphT *graph, uint32_t node_index, int batch_position) { in CheckStridedSlice()
278 bool CheckGruCell(schema::MetaGraphT *graph, uint32_t node_index) { in CheckGruCell()
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dset_fracz_group_attr.cc67 for (const auto &node_index : manager->node_users()[node]) { in GetOutputItem() local
169 [](const KernelWithIndex &node_index) { in GetCNodeNeighborFraczNodes()
184 [](const KernelWithIndex &node_index) { in GetNeighborFraczNodes()
200 [](const KernelWithIndex &node_index) { in GetNeighborFraczNodes()
221 KernelWithIndex node_index = todo.back(); in SetAttrFraczGroup() local
245 KernelWithIndex node_index = todo.back(); in SetAttrFraczGroup() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dconv2d_delegate_fp32_coder.cc67 … const LiteGraph::Node *node, size_t node_index, in CPUConvolutionFP32CoderSelect()
106size_t node_index, Target target, int schema_version) { in CreateDelegateConv()
112 … const LiteGraph::Node *node, size_t node_index, Target target, in CPUConvDwFp32CoderCreator()
120 … const LiteGraph::Node *node, size_t node_index, in CPUConv2DFusionFP32CoderCreator()
Dgroupnorm_fp32_coder.h27 const LiteGraph::Node *node, size_t node_index, Target target) in GroupNormFP32Coder()
Dgather_dynamic_fp32_coder.h27 const LiteGraph::Node *node, size_t node_index, Target target) in GatherDynamicFP32Coder()
Dsoftmax_fp32_coder.h25 const LiteGraph::Node *node, size_t node_index, Target target) in SoftMaxFP32Coder()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dconv2d_delegate_dynamic_fp16_coder.cc81 … const LiteGraph::Node *node, size_t node_index, in CPUConvFP16DynamicCoderSelect()
117 … const LiteGraph::Node *node, size_t node_index, Target target, in CreateConvDelegateFp16()
125 … const LiteGraph::Node *node, size_t node_index, in CPUConv2DFusionDynamicFP16CoderCreator()
Dconv2d_delegate_fp16_coder.cc110 … const LiteGraph::Node *node, size_t node_index, in CPUConvolutionFP16CoderSelect()
157 … const LiteGraph::Node *node, size_t node_index, Target target, in CreateDelegateFp16Conv()
164 … const LiteGraph::Node *node, size_t node_index, Target target, in CPUConvDwFp16CoderCreator()
223 … const LiteGraph::Node *node, size_t node_index, in CPUConv2DFusionFP16CoderCreator()
Dactivation_fp16_coder.h27 const LiteGraph::Node *node, size_t node_index, Target target) in ActivationFP16Coder()
Dconcat_dynamic_fp16_coder.h28 const LiteGraph::Node *node, size_t node_index, Target target) in ConcatDynamicFP16Coder()
Dscale_fp16_coder.h30 const LiteGraph::Node *node, size_t node_index, Target target) in ScaleFP16Coder()
Dcustom_gru_fp16_coder.h29 const LiteGraph::Node *node, size_t node_index, Target target) in CustomGruFP16Coder()
Dtranspose_fp16_coder.h27 const LiteGraph::Node *node, size_t node_index, Target target) in TransposeFp16Coder()
Darithmetic_self_fp16_coder.h93 const LiteGraph::Node *node, size_t node_index, Target target) in ArithmeticSelfFP16Coder()
Ddeconv2d_fp16_coder.h30 const LiteGraph::Node *node, size_t node_index, Target target) in DeConvolutionFP16Coder()
Dconv_depthwise_3x3_fp16_coder.h30 const LiteGraph::Node *node, size_t node_index, Target target) in ConvolutionDepthwise3x3FP16Coder()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_parse_graph.cc224 void Eliminate_Aux(size_t node_index, const std::shared_ptr<Graph> &graph, in Eliminate_Aux()
276 void EliminateAuxOutgoingInput(size_t node_index, const std::shared_ptr<Graph> &graph, size_t i) { in EliminateAuxOutgoingInput()
347 void EliminateAuxOutgoingAuxInput(size_t node_index, const std::shared_ptr<Graph> &graph, size_t i)… in EliminateAuxOutgoingAuxInput()
428 void Eliminate_Aux_Outgoing(size_t node_index, const std::shared_ptr<Graph> &graph) { in Eliminate_Aux_Outgoing()
453 for (size_t node_index = 0; node_index < graph->nodes.size(); node_index++) { in EliminateGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/base/
Ddetection_post_process_base_coder.h32 const LiteGraph::Node *node, size_t node_index, Target target) in DetectionPostProcessBaseCoder()
Dfull_connection_base_coder.h28 const LiteGraph::Node *node, size_t node_index, Target target) in FullConnectionBaseCoder()
Dreshape_base_coder.h27 const LiteGraph::Node *node, size_t node_index, Target target) in ReshapeBaseCoder()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/cmsis-nn/int8/
Dpooling_int8_coder.h30 const LiteGraph::Node *node, size_t node_index, Target target) in PoolingInt8Coder()
Dfullconnection_int8_coder.h29 const LiteGraph::Node *node, size_t node_index, Target target) in FullConnectionInt8Coder()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32_grad/
Dassign_coder.h27 const LiteGraph::Node *node, size_t node_index, Target target) in AssignCoder()

123456789