Searched defs:tuple_input (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/python/mutable/ |
| D | test_const_arg_tensor.py | 258 def net_func_tuple_input(tuple_input): argument
275 def net_func_tuple_input(tuple_input): argument
292 def construct(self, tuple_input): argument
311 def net_func_tuple_input(tuple_input): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | concat_op_pass.cc | 79 int64_t SplitTupleInputs(const FuncGraphPtr &graph, const AnfNodePtr &tuple_input, in SplitTupleInputs()
|
| D | grouped_matmul_op_pass.cc | 125 …4_t GroupedMatmulOpPass::SplitTupleInputs(const FuncGraphPtr &graph, const AnfNodePtr &tuple_input, in SplitTupleInputs()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | insert_format_transform_op.cc | 40 auto tuple_input = std::make_shared<ValueNode>(tuple_value); in CreateTupleValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/ |
| D | do_pyboost_cast.h | 92 const TupleInput &tuple_input, in GetTypeInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | dropout_unify_mindir.cc | 448 auto tuple_input = mask_input_cnode->input(kIndex1); in BuildDropoutDoMask() local
755 auto tuple_input = mask_input_cnode->input(kIndex1); in Process() local
|
| D | all_to_all_unify_mindir.cc | 107 auto tuple_input = neighbor_exchange->input(1); in CreateAllToAllvNode() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/fallback/ |
| D | test_graph_fallback_runtime.py | 240 def __init__(self, tuple_input): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | insert_type_transform_op.cc | 50 int64_t SplitTupleInputsForInsertType(const FuncGraphPtr &graph, const AnfNodePtr &tuple_input, in SplitTupleInputsForInsertType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_splitter.cc | 145 const auto tuple_input = real_output->cast<CNodePtr>()->input(i + kSizeOne); in CreateReplacedOutputNode() local
1428 auto tuple_input = common::AnfAlgo::GetInputNode(tuple_get_item_node, kIndex0); in RecursiveSetTupeGetItemLabel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/ |
| D | helper.cc | 1392 int64_t SplitTupleInputs(const FuncGraphPtr &graph, const AnfNodePtr &tuple_input, in SplitTupleInputs()
|