Home
last modified time | relevance | path

Searched refs:tuple_inputs (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dcall_graph_tuple_transform.h188 auto tuple_inputs = node->cast<CNodePtr>()->inputs(); in TransformLayerNode() local
190 for (size_t i = 1; i < tuple_inputs.size(); i++) { in TransformLayerNode()
191 if (!IsValueNode<FuncGraph>(tuple_inputs[i])) { in TransformLayerNode()
195 FuncGraphPtr fg = GetValueNode<FuncGraphPtr>(tuple_inputs[i]); in TransformLayerNode()
198 tuple_inputs[i] = NewValueNode(transformed_fg); in TransformLayerNode()
203 *trans_node = node->func_graph()->NewCNode(tuple_inputs); in TransformLayerNode()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc65 auto tuple_inputs = cnode->inputs(); in GetShapeVectorFromCNode() local
66 MS_ASSERT(tuple_inputs.size() == kTupleGetItemInputSize); in GetShapeVectorFromCNode()
67 auto get_item_input_cnode = tuple_inputs.at(kSecondIndex); in GetShapeVectorFromCNode()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc96 auto tuple_inputs = switch_layer_inputs[kSwitchLayerBranchPos]->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchLayerNode() local
99 for (size_t i = kMakeTupleInputStartPos; i < tuple_inputs.size(); ++i) { in FetchParameterBySwitchLayerNode()
100 if (AnfAlgo::CheckPrimitiveType(tuple_inputs[i], prim::kPrimPartial)) { in FetchParameterBySwitchLayerNode()
102 const auto &func_graph = GetFuncGraphFromPartial(tuple_inputs[i]); in FetchParameterBySwitchLayerNode()
104 const auto &partial_inputs = tuple_inputs[i]->cast<CNodePtr>()->inputs(); in FetchParameterBySwitchLayerNode()
120 } else if (tuple_inputs[i]->isa<ValueNode>() && IsValueNode<FuncGraph>(tuple_inputs[i])) { in FetchParameterBySwitchLayerNode()
122 const auto &func_graph = GetValueNode<FuncGraphPtr>(tuple_inputs[i]); in FetchParameterBySwitchLayerNode()
534 const auto tuple_inputs = switch_layer_inputs[kSwitchLayerBranchPos]->cast<CNodePtr>()->inputs(); in IsSubCallNode() local
535 if (tuple_inputs.size() <= kMakeTupleInputStartPos) { in IsSubCallNode()
541 if (AnfAlgo::CheckPrimitiveType(tuple_inputs[kMakeTupleInputStartPos], prim::kPrimPartial)) { in IsSubCallNode()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_auto_monad.cc1189 AnfNodePtrList tuple_inputs; in LinkArguments() local
1190 tuple_inputs.reserve(args.size() + 1); in LinkArguments()
1191 tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in LinkArguments()
1196 tuple_inputs.emplace_back(value); in LinkArguments()
1204 (void)tuple_inputs.emplace_back(AssignAll(target, value, true, keep, false)); in LinkArguments()
1206 auto new_tuple = kernel_graph_->NewCNode(tuple_inputs); in LinkArguments()
1209 …(void)std::transform(tuple_inputs.begin() + 1, tuple_inputs.end(), std::back_inserter(element_abst… in LinkArguments()
1262 AnfNodePtrList tuple_inputs; in AssignAll() local
1263 tuple_inputs.reserve(targets.size() + 1); in AssignAll()
1264 tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in AssignAll()
[all …]
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dauto_monad.cc1175 std::vector<AnfNodePtr> tuple_inputs; in HandleNoEliminateNodes() local
1177 tuple_inputs.reserve(no_eliminate_nodes_.size() + 1); in HandleNoEliminateNodes()
1179 tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in HandleNoEliminateNodes()
1181 tuple_inputs.emplace_back(node); in HandleNoEliminateNodes()
1184 auto make_tuple_node = func_graph_->NewCNode(tuple_inputs); in HandleNoEliminateNodes()
Dstatic_analysis.cc700 auto tuple_inputs = tuple_node->cast<CNodePtr>()->inputs(); in JoinBranchesFailedInfo() local
701 for (size_t i = 1; i < tuple_inputs.size(); i++) { in JoinBranchesFailedInfo()
702 buffer << ", branch" << i << ": " << tuple_inputs.at(i); in JoinBranchesFailedInfo()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc410 void PlantTensorTupleToVector(const py::tuple &tuple_inputs, const PrimitivePtr &op_prim, in PlantTensorTupleToVector() argument
414 for (const auto &input_object : tuple_inputs) { in PlantTensorTupleToVector()
422 …op_prim->set_attr(kAttrDynInputSizes, MakeValue(std::vector<int64_t>{SizeToLong(tuple_inputs.size(… in PlantTensorTupleToVector()
448 auto tuple_inputs = py::cast<py::tuple>(input_object); in ConvertMultiPyObjectToTensor() local
449 if (tuple_inputs.empty()) { in ConvertMultiPyObjectToTensor()
452 if (py::isinstance<tensor::Tensor>(tuple_inputs[0])) { in ConvertMultiPyObjectToTensor()
453 PlantTensorTupleToVector(tuple_inputs, op_prim, input_tensors); in ConvertMultiPyObjectToTensor()
479 py::tuple tuple_inputs(list_inputs.size()); in ConvertPyObjectToTensor() local
480 for (size_t i = 0; i < tuple_inputs.size(); ++i) { in ConvertPyObjectToTensor()
481 tuple_inputs[i] = list_inputs[i]; in ConvertPyObjectToTensor()
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc595 auto tuple_inputs = input_cnode->inputs(); in GetCNodeInputAbstract() local
596 MS_ASSERT(tuple_inputs.size() == kTupleGetItemInputSize); in GetCNodeInputAbstract()
597 auto get_item_input_cnode = tuple_inputs.at(1); in GetCNodeInputAbstract()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc1668 auto tuple_inputs = node->cast<CNodePtr>()->inputs(); in GetRealOpNode() local
1669 if (tuple_inputs.size() < LongToSize(index + 1L)) { in GetRealOpNode()
1670 MS_LOG(ERROR) << "make tuple input items node not correct! size:" << tuple_inputs.size() in GetRealOpNode()
1675 return GetRealOpNode(tuple_inputs[LongToSize(index + 1L)]); in GetRealOpNode()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc555 void PlantTensorTupleToVector(const py::tuple &tuple_inputs, std::vector<tensor::TensorPtr> *tensor… in PlantTensorTupleToVector() argument
557 for (const auto &input_object : tuple_inputs) { in PlantTensorTupleToVector()