| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_conv_info.h | 74 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
85 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
97 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
108 : MultiConvSplit(strategy, primitive_type, fmk_type, num) {} in MultiConvSplit() argument
|
| D | parallel_pass.h | 35 …rallelPass(const std::unordered_map<std::string, SplitStrategy> &strategys, const int32_t fmk_type) in ParallelPass()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | node_infershape.h | 36 : fmk_type_(fmk_type), train_flag_(train_flag) {} in fmk_type_() argument
38 void Init(FmkType fmk_type, bool train_flag) { in Init()
|
| D | transpose_strategy.h | 35 void Init(FmkType fmk_type, bool train_flag) { in Init()
|
| /third_party/mindspore/mindspore/lite/tools/converter/registry/ |
| D | node_parser_registry.cc | 28 NodeParserRegistry::NodeParserRegistry(converter::FmkType fmk_type, const std::string &node_type, in NodeParserRegistry()
34 converter::NodeParserPtr NodeParserRegistry::GetNodeParser(converter::FmkType fmk_type, const std::… in GetNodeParser()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_pool_parser.cc | 96 int fmk_type = converter::FmkType::kFmkTypeOnnx; in Parse() local
168 int fmk_type = converter::FmkType::kFmkTypeOnnx; in Parse() local
|
| D | onnx_slice_parser.cc | 93 int fmk_type = converter::FmkType::kFmkTypeOnnx; in Parse() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_pooling_parser.cc | 144 int fmk_type = converter::FmkType::kFmkTypeCaffe; in Parse() local
157 int fmk_type = converter::kFmkTypeCaffe; in Parse() local
|
| D | caffe_batchnorm_parser.cc | 57 int fmk_type = converter::FmkType::kFmkTypeCaffe; in Parse() local
|
| D | caffe_deconvolution_parser.cc | 93 int fmk_type = converter::FmkType::kFmkTypeCaffe; in Parse() local
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | fetch_content.cc | 114 …orValue(const ValueNodePtr &value_node, const PrimitivePtr &primitive, converter::FmkType fmk_type, in FetchFromTensorValue()
234 int FetchFromDefaultParam(const ParameterPtr ¶m_node, const converter::FmkType &fmk_type, DataI… in FetchFromDefaultParam()
270 int FetchDataFromParameterNode(const CNodePtr &cnode, size_t index, converter::FmkType fmk_type, bo… in FetchDataFromParameterNode()
300 int FetchDataFromValueNode(const CNodePtr &cnode, size_t index, converter::FmkType fmk_type, bool t… in FetchDataFromValueNode()
337 int SetFormatForCnode(const CNodePtr &cnode, size_t index, converter::FmkType fmk_type, bool train_… in SetFormatForCnode()
369 int FetchDataFromCNode(const CNodePtr &cnode, size_t index, converter::FmkType fmk_type, bool train… in FetchDataFromCNode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | unify_format.h | 28 : ToFormatBase(fmk_type, train_flag) {} in ToFormatBase() argument
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | mindir_control_flow_adjust.h | 33 void SetFmkType(FmkType fmk_type) { fmk_type_ = fmk_type; } in SetFmkType()
|
| D | mindir_adjust.h | 32 void SetFmkType(FmkType fmk_type) { fmk_type_ = fmk_type; } in SetFmkType()
|
| D | primitive_adjust.h | 70 void SetFmkType(FmkType fmk_type) { fmk_type_ = fmk_type; } in SetFmkType()
|
| /third_party/mindspore/mindspore/lite/include/registry/ |
| D | node_parser_registry.h | 53 #define REG_NODE_PARSER(fmk_type, node_type, node_parser) \ argument
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | maxpool_fusion_mapper.cc | 34 int fmk_type = attr_val != nullptr ? GetValue<int>(attr_val) : converter::kFmkTypeTf; in Mapper() local
|
| D | stridedslice_mapper.cc | 35 int fmk_type = attr_val != nullptr ? GetValue<int>(attr_val) : converter::kFmkTypeTf; in Mapper() local
|
| D | batchnorm_mapper.cc | 34 int fmk_type = attr_val != nullptr ? GetValue<int>(attr_val) : converter::kFmkTypeTf; in Mapper() local
|
| D | avgpool_fusion_mapper.cc | 34 int fmk_type = attr_val != nullptr ? GetValue<int>(attr_val) : converter::kFmkTypeTf; in Mapper() local
|
| D | conv2d_transpose_fusion_mapper.cc | 34 int fmk_type = attr_val != nullptr ? GetValue<int>(attr_val) : converter::kFmkTypeTf; in Mapper() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/ |
| D | acl_pass.h | 36 explicit AclPass(FmkType fmk_type) : Pass("Acl"), fmk_type_(fmk_type) {} in AclPass()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.h | 44 explicit InferShapePass(converter::FmkType fmk_type) : fmk_type_(fmk_type) {} in InferShapePass()
|
| /third_party/mindspore/mindspore/lite/tools/converter/ |
| D | converter_context.h | 55 void set_fmk_type(const std::string &fmk_type) { fmk_type_ = fmk_type; } in set_fmk_type()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_control_op_pass.cc | 28 …nctionalizeControlOpPass::NewFuncGraph(const std::string &subgraph_name, const FmkType &fmk_type) { in NewFuncGraph()
|