Home
last modified time | relevance | path

Searched defs:node_type (Results 1 – 25 of 88) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/lite/include/registry/
Dnode_parser_registry.h55 NodeParserRegistry::NodeParserRegistry(converter::FmkType fmk_type, const std::string &node_type, in NodeParserRegistry()
59 …rPtr NodeParserRegistry::GetNodeParser(converter::FmkType fmk_type, const std::string &node_type) { in GetNodeParser()
68 #define REG_NODE_PARSER(fmk_type, node_type, node_parser) \ argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/registry/
Dnode_parser_registry.cc30 …ParserRegistry::NodeParserRegistry(converter::FmkType fmk_type, const std::vector<char> &node_type, in NodeParserRegistry()
44 const std::vector<char> &node_type) { in GetNodeParser()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/utils/
Dtrain_utils.cc48 auto node_type = static_cast<schema::PrimitiveType>(GetPrimitiveType(node->primitive_)); in IsOptimizer() local
57 auto node_type = static_cast<schema::PrimitiveType>(GetPrimitiveType(node->primitive_)); in IsMaskOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/
Dstack_nnapi.cc47 OperationCode node_type = ANEURALNETWORKS_RESHAPE; in AddOpToNNAPIModel() local
87 OperationCode node_type = ANEURALNETWORKS_CONCATENATION; in AddOpToNNAPIModel() local
Dtranspose_nnapi.cc30 OperationCode node_type = ANEURALNETWORKS_TRANSPOSE; in AddOpToNNAPIModel() local
Dtopk_nnapi.cc35 OperationCode node_type = ANEURALNETWORKS_TOPK_V2; in AddOpToNNAPIModel() local
Dstrided_slice_nnapi.cc38 OperationCode node_type = ANEURALNETWORKS_STRIDED_SLICE; in AddOpToNNAPIModel() local
Dreshape_nnapi.cc32 OperationCode node_type = ANEURALNETWORKS_RESHAPE; in AddOpToNNAPIModel() local
Dpadding_nnapi.cc42 OperationCode node_type = const_value_ == 0 ? ANEURALNETWORKS_PAD : ANEURALNETWORKS_PAD_V2; in AddOpToNNAPIModel() local
Dgather_nnapi.cc38 OperationCode node_type = ANEURALNETWORKS_GATHER; in AddOpToNNAPIModel() local
Dconcat_nnapi.cc41 OperationCode node_type = ANEURALNETWORKS_CONCATENATION; in AddOpToNNAPIModel() local
Dcast_nnapi.cc35 OperationCode node_type = ANEURALNETWORKS_CAST; in AddOpToNNAPIModel() local
Dactivation_nnapi.cc51 OperationCode node_type = activation_type.at(act_type_); in AddOpToNNAPIModel() local
Dreduce_nnapi.cc44 OperationCode node_type; in AddOpToNNAPIModel() local
Dsoftmax_nnapi.cc43 OperationCode node_type = ANEURALNETWORKS_SOFTMAX; in AddOpToNNAPIModel() local
Dinstance_norm_nnapi.cc48 OperationCode node_type = ANEURALNETWORKS_INSTANCE_NORMALIZATION; in AddOpToNNAPIModel() local
Dfull_connection_nnapi.cc40 OperationCode node_type = ANEURALNETWORKS_FULLY_CONNECTED; in AddOpToNNAPIModel() local
Dscale_nnapi.cc44 OperationCode node_type = ANEURALNETWORKS_MUL; in AddOpToNNAPIModel() local
Darithmetic_nnapi.cc72 OperationCode node_type = arithmetic_type.at(type_); in AddOpToNNAPIModel() local
Dsplit_nnapi.cc58 OperationCode node_type = ANEURALNETWORKS_SPLIT; in AddOpToNNAPIModel() local
Dresize_nnapi.cc69 OperationCode node_type = method_ == static_cast<int>(schema::ResizeMethod_LINEAR) in AddOpToNNAPIModel() local
Dconv_transpose_nnapi.cc55 OperationCode node_type = ANEURALNETWORKS_TRANSPOSE_CONV_2D; in AddOpToNNAPIModel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/
Dpytorch_node_parser.cc32 std::string node_type = kind.toUnqualString(); in GetTorchNodeType() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dtensor_category.cc22 Category TensorCategory(const int node_type, const size_t shape_num, const TypeId data_type, const … in TensorCategory()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Ddebug_info_manager.h62 std::string node_type; member
81 std::string node_type; member

1234