Home
last modified time | relevance | path

Searched refs:ValueTuple (Results 1 – 25 of 85) sorted by relevance

1234

/third_party/mindspore/mindspore/core/ir/
Dvalue.h72 class MS_CORE_API ValueTuple : public ValueSequeue {
74 explicit ValueTuple(const std::vector<ValuePtr> &elements) : ValueSequeue(elements) {} in ValueTuple() function
75 ValueTuple(const std::initializer_list<ValuePtr> &elements) : ValueSequeue(elements) {} in ValueTuple() function
76 ~ValueTuple() override = default;
77 MS_DECLARE_PARENT(ValueTuple, ValueSequeue)
83 using ValueTuplePtr = std::shared_ptr<ValueTuple>;
98 inline ValuePtr MakeValue(const std::vector<ValuePtr> &v) { return std::make_shared<ValueTuple>(v);… in MakeValue()
99 inline ValuePtr MakeValue(std::initializer_list<ValuePtr> v) { return std::make_shared<ValueTuple>(… in MakeValue()
110 return std::make_shared<ValueTuple>(list);
Dvalue_extends.cc44 abstract::AbstractBasePtr ValueTuple::ToAbstract() { in ToAbstract()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter_util.h49 if (!value->isa<ValueTuple>() && !value->isa<ValueList>()) { in ConvertAnyUtil()
53 …auto vec = value->isa<ValueTuple>() ? value->cast<ValueTuplePtr>()->value() : value->cast<ValueLis… in ConvertAnyUtil()
Dop_adapter_util.cc134 …auto vec = value->isa<ValueTuple>() ? value->cast<ValueTuplePtr>()->value() : value->cast<ValueLis… in VectorToTensorUtil()
185 } else if (value->isa<ValueList>() || value->isa<ValueTuple>()) { in ConvertAnyUtil()
/third_party/mindspore/mindspore/core/ops/
Dgetnext.cc36 …std::vector<ValuePtr> shape = shape_attr->isa<ValueTuple>() ? shape_attr->cast<ValueTuplePtr>()->v… in GetShapeVector()
39 std::vector<ValuePtr> shape_elements_list = shape_elements->isa<ValueTuple>() in GetShapeVector()
Dreduce_sum.cc40 if (axis->isa<ValueTuple>() || axis->isa<ValueList>()) { in InferImplReduceFuncCalShape()
42 …axis->isa<ValueTuple>() ? axis->cast<ValueTuplePtr>()->value() : axis->cast<ValueListPtr>()->value… in InferImplReduceFuncCalShape()
162 axis_value = std::make_shared<ValueTuple>(value_list); in InferShape()
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_addn.cc40 auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2}); in TEST_F()
73 …auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2, tensor… in TEST_F()
Dtest_ops_concat.cc43 …auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2, tensor… in TEST_F()
77 …auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2, tensor… in TEST_F()
Dtest_ops_assert.cc54 auto inputs = std::make_shared<ValueTuple>(inputs_); in TEST_F()
85 auto inputs = std::make_shared<ValueTuple>(inputs_); in TEST_F()
Dtest_ops_pack.cc43 auto input_ = std::make_shared<ValueTuple>(std::vector<ValuePtr>{input0, input1}); in TEST_F()
Dtest_ops_merge.cc43 auto input = std::make_shared<ValueTuple>(inputs_); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_input_to_tensor_input.cc40 } else if (value->isa<ValueTuple>()) { in CreateTensorInput()
78 if (IsValueNode<Scalar>(input_node) || IsValueNode<ValueTuple>(input_node)) { in ConstInputToTensorInput()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dreduce_eliminate.h77 auto new_shape = std::make_shared<ValueTuple>(elements); in operator()
137 if (IsValueNode<ValueTuple>(vnode)) { in Visit()
Dspecial_op_eliminate.h460 if (!value->isa<ValueTuple>()) { in FillGetItem()
487 if (value->isa<ValueTuple>()) { in FillZero()
493 out = std::make_shared<ValueTuple>(value_list); in FillZero()
562 if (!value_node->value()->isa<ValueTuple>()) { in OperatorHandle4()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/
Dboundingbox_encode_gpu_kernel.h105 if (means->isa<ValueTuple>() || means->isa<ValueList>()) { in Init()
117 if (stds->isa<ValueTuple>() || stds->isa<ValueList>()) { in Init()
Dboundingbox_decode_gpu_kernel.h106 if (means->isa<ValueTuple>() || means->isa<ValueList>()) { in Init()
119 if (stds->isa<ValueTuple>() || stds->isa<ValueList>()) { in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dboundingbox_encode_cpu_kernel.cc31 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<ValueTuple>() || in InitKernel()
43 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<ValueTuple>() || in InitKernel()
Dboundingbox_decode_cpu_kernel.cc31 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<ValueTuple>() || in InitKernel()
43 if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<ValueTuple>() || in InitKernel()
/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils.cc281 if (value->isa<ValueTuple>()) { in TensorValueToTensor()
290 } else if (element->isa<ValueTuple>()) { in TensorValueToTensor()
308 } else if (value->isa<ValueTuple>()) { in CountValueNum()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc88 } else if (IsValueNode<ValueTuple>(node)) { in GetValueListShape()
108 if (IsValueNode<ValueList>(node) || IsValueNode<ValueTuple>(node)) { in GetNodeShape()
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc143 ValueTuplePtr var = std::make_shared<ValueTuple>(elements); in Make_Manager()
162 ValueTuplePtr var2 = std::make_shared<ValueTuple>(elements2); in Make_Manager()
187 ValueTuplePtr var_t = std::make_shared<ValueTuple>(elements_t); in Make_Manager()
195 ValueTuplePtr var_t2 = std::make_shared<ValueTuple>(elements_t2); in Make_Manager()
228 ValueTuplePtr strategy_tuple = std::make_shared<ValueTuple>(elements); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc230 if (elem->isa<ValueTuple>() || elem->isa<ValueList>()) { in IsAllFuncInValueSequence()
250 if (elem->isa<ValueTuple>() || elem->isa<ValueList>()) { in TransformToMakeTupleNodes()
305 if (IsValueNode<ValueTuple>(resolved_node) || IsValueNode<ValueList>(resolved_node)) { in ResolveObjectAndAddToManager()
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Ddata_test.cc77 …ASSERT_EQ(*func_tuple_built, ValueTuple(std::vector<ValuePtr>{prim::kPrimReturn, prim::kPrimScalar… in TEST_F()
87 …ASSERT_EQ(*func_tuple_built, ValueTuple(std::vector<ValuePtr>{std::make_shared<Int64Imm>(1), prim:… in TEST_F()
/third_party/mindspore/tests/ut/cpp/ir/
Dvalue_test.cc75 ValuePtr tv = std::make_shared<ValueTuple>(std::vector<ValuePtr>({boolv, i64v, f32v, sv, vv})); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc68 (IsValueNode<ValueList>(node_inputs[1]) || IsValueNode<ValueTuple>(node_inputs[1]))) { in ExtractInputParameterByNode()
158 (IsValueNode<ValueList>(node_inputs[1]) || IsValueNode<ValueTuple>(node_inputs[1]))) { in ExtractInputTypeLengthByNode()

1234