| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_node_split.h | 34 virtual AnfNodePtr DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) = 0;
45 AnfNodePtr DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) override;
|
| D | multi_node_split.cc | 47 AnfNodePtr MultiNodeSplitProxy::DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) { in DoSplit() function in mindspore::opt::MultiNodeSplitProxy
54 auto res_node = multi_node_split_->DoSplit(func_graph, node); in DoSplit()
|
| D | operator_info.h | 59 int DoSplit();
|
| D | parallel_pass.cc | 134 if (parallel_operator->DoSplit() == RET_ERROR) { in Run()
|
| D | multi_conv_info.h | 33 AnfNodePtr DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) override;
|
| D | operator_info.cc | 189 int OperatorInfo::DoSplit() { in DoSplit() function in mindspore::opt::OperatorInfo
|
| D | multi_conv_info.cc | 243 AnfNodePtr MultiConvSplit::DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) { in DoSplit() function in mindspore::opt::MultiConvSplit
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | transdata_split.cc | 36 return DoSplit(func_graph, node); in Process()
59 CNodePtr TransDataSplit::DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) const { in DoSplit() function in mindspore::opt::TransDataSplit
|
| D | transdata_split.h | 41 CNodePtr DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) const;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | split_base.h | 26 int DoSplit(const void *in_data, void **out_data, const int *input_shape, int offset, int num_unit,
|
| D | split_base.c | 22 int DoSplit(const void *in_data, void **out_data, const int *input_shape, int offset, int num_unit, in DoSplit() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | transposed_update_fusion.h | 43 CNodePtr DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &node) const;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | deal_ref_and_split_unsupported_transdata.cc | 282 return DoSplit(func_graph, cnode); in SplitTransdataIfNotSupported()
305 auto after_split_node = DoSplit(func_graph, cnode); in SplitTransdataIfNotSupported()
309 return DoSplit(func_graph, next_trans_node); in SplitTransdataIfNotSupported()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | multi_conv_split_pass.cc | 92 return multi_node_split_proxy->DoSplit(func_graph, node); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | split_cpu_kernel.cc | 83 …(void)DoSplit(input, reinterpret_cast<void **>(output), &input_shape_[0], SizeToInt(start), SizeTo… in LaunchSplit()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | glu_fp32.cc | 108 …DoSplit(input_ptr_, reinterpret_cast<void **>(split_ptr_.data()), in_tensors_.front()->shape().dat… in Split()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | split_base.cc | 121 …auto ret = DoSplit(input_ptr_, output_ptr_.data(), in_tensors_.front()->shape().data(), thread_off… in Split()
|