Home
last modified time | relevance | path

Searched refs:SplitStrategy (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dmulti_conv_info.h27 …explicit MultiConvSplit(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_t…
57 SplitStrategy strategy_{};
73 …MultiConvSplitN(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type = -1…
84 …MultiConvSplitCIN(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type = …
95 …MultiConvSplitCOUT(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type =…
107 …MultiConvSplitH(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t fmk_type = -1…
Doperator_info.h49 OperatorInfo(const std::string &name, const SplitStrategy &strategy) in OperatorInfo()
79 virtual int CheckStrategy(const SplitStrategy &strategy) = 0;
83 SplitStrategy strategy_;
98 std::unique_ptr<OperatorInfo> OperatorInfoCreator(const std::string &name, const SplitStrategy &str… in OperatorInfoCreator()
Dparallel_pass.h35 …explicit ParallelPass(const std::unordered_map<std::string, SplitStrategy> &strategys, const int32… in ParallelPass()
54 std::unordered_map<std::string, SplitStrategy> split_strategys_;
Dmulti_node_split.h39 …explicit MultiNodeSplitProxy(const SplitStrategy &strategy, PrimitiveType primitive_type, int32_t …
53 SplitStrategy strategy_{};
Dconv2d_info.h33 …Conv2DInfo(const std::string &name, const SplitStrategy &strategy) : OperatorInfo(name, strategy) … in Conv2DInfo()
37 int CheckStrategy(const SplitStrategy &strategy) override;
Ddepthwise_conv2d_info.h30 …DepthwiseConv2DInfo(const std::string &name, const SplitStrategy &strategy) : Conv2DInfo(name, str… in DepthwiseConv2DInfo()
36 int CheckStrategy(const SplitStrategy &strategy) override;
Dsplit_strategy.h73 struct SplitStrategy { struct
106 std::unordered_map<std::string, opt::SplitStrategy> ParserSplitStrategy(
Dsplit_strategy.cc46 std::unordered_map<std::string, opt::SplitStrategy> ParserSplitStrategy(const std::vector<int64_t> … in ParserSplitStrategy()
49 std::unordered_map<std::string, opt::SplitStrategy> split_strategys; in ParserSplitStrategy()
Doperator_info_register.h29 …std::function<std::unique_ptr<opt::OperatorInfo>(const std::string &name, const SplitStrategy &str…
Dconv2d_info.cc40 int Conv2DInfo::CheckStrategy(const SplitStrategy &strategy) { in CheckStrategy()
Ddepthwise_conv2d_info.cc127 int DepthwiseConv2DInfo::CheckStrategy(const SplitStrategy &strategy) { in CheckStrategy()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dmulti_conv_split_pass.h36 …explicit MultiConvSplitPass(std::unordered_map<std::string, SplitStrategy> strategys, int32_t fmk_…
50 std::unordered_map<std::string, SplitStrategy> strategys_{};
/third_party/mindspore/mindspore/lite/tools/converter/
Danf_transform.cc243 std::unordered_map<std::string, opt::SplitStrategy> split_strategys = in RunParallelPass()