Home
last modified time | relevance | path

Searched refs:ValueTuplePtr (Results 1 – 25 of 73) sorted by relevance

123

/third_party/mindspore/mindspore/core/ops/
Dgetnext.cc36 …std::vector<ValuePtr> shape = shape_attr->isa<ValueTuple>() ? shape_attr->cast<ValueTuplePtr>()->v… in GetShapeVector()
40 ? shape_elements->cast<ValueTuplePtr>()->value() in GetShapeVector()
Ddropout_gen_mask.cc124 : shape_max->cast<ValueTuplePtr>()->value(); in InferShape()
129 : shape_min->cast<ValueTuplePtr>()->value(); in InferShape()
Dneighborexchange.cc51 if (attr->cast<ValueTuplePtr>() == nullptr) { in CheckAttr()
67 if (attr->cast<ValueTuplePtr>() != nullptr) { in CheckAttr()
Dtranspose.cc37 auto perm_val = perm->cast<ValueTuplePtr>(); in InferShape()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dslice_info.cc34 ValueTuplePtr value_tuple = input_value->cast<ValueTuplePtr>(); in GetInput()
207 auto value_ptr = MakeValue(value_tuple)->cast<ValueTuplePtr>(); in CreateValueTupleAndNodePtr()
Dactivation_info.cc145 ValueTuplePtr value_tuple = iter->second->cast<ValueTuplePtr>(); in GetAttrs()
493 Status SqueezeInfo::InferAxis(const ValueTuplePtr &value_tuple) { in InferAxis()
509 axis_ = MakeValue(axis)->cast<ValueTuplePtr>(); in InferAxis()
523 axis_ = MakeValue(axis)->cast<ValueTuplePtr>(); in InferAxis()
534 auto value_tuple = iter->second->cast<ValueTuplePtr>(); in GetAttrs()
Dtranspose_info.cc58 ValueTuplePtr dim_tuple = input_value_[1]->cast<ValueTuplePtr>(); in ComputeAxis()
Duniform_real_info.cc42 ValueTuplePtr shape_value = input_value_[0]->cast<ValueTuplePtr>(); in GetAttrs()
Dtile_info.cc43 ValueTuplePtr multiples = input_value_[1]->cast<ValueTuplePtr>(); in GetAttrs()
Dstrided_slice_info.cc50 ValueTuplePtr value_tuple = input_value->cast<ValueTuplePtr>(); in GetInput()
Dactivation_info.h241 Status InferAxis(const ValueTuplePtr &value_tuple);
248 ValueTuplePtr axis_;
Dgather_v2_p_info.cc40 if (manual_split_without_offset_iter->second->cast<ValueTuplePtr>() == nullptr) { in GetManualSplitWithoutOffsetAttr()
44 …<ValuePtr> value_vector = manual_split_without_offset_iter->second->cast<ValueTuplePtr>()->value(); in GetManualSplitWithoutOffsetAttr()
75 auto var = manual_split_with_offset_iter->second->cast<ValueTuplePtr>(); in GetManualSplitAttr()
87 std::vector<ValuePtr> value_vector = ele->cast<ValueTuplePtr>()->value(); in GetManualSplitAttr()
Dbatch_parallel_info.cc121 auto shape_ptr = input_value_[0]->cast<ValueTuplePtr>(); in GetAttrs()
Dreshape_info.cc91 ValueTuplePtr dim_tuple = input_value_[1]->cast<ValueTuplePtr>(); in GetParameterInput()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter_util.cc63 auto vec = value->cast<ValueTuplePtr>(); in ConvertAnyUtil()
81 auto vec = value->cast<ValueTuplePtr>(); in ConvertAnyUtil()
95 auto vec = value->cast<ValueTuplePtr>(); in ConvertAnyUtil()
134 …auto vec = value->isa<ValueTuple>() ? value->cast<ValueTuplePtr>()->value() : value->cast<ValueLis… in VectorToTensorUtil()
Dop_adapter_util.h53 …auto vec = value->isa<ValueTuple>() ? value->cast<ValueTuplePtr>()->value() : value->cast<ValueLis… in ConvertAnyUtil()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc231 auto x_shp_data = x_shp_value->cast<ValueTuplePtr>()->value(); in DoInferReduceShape()
261 auto arg_x_value = arg_x->BuildValue()->cast<ValueTuplePtr>(); in InferImplBroadcastGradientArgs()
264 auto arg_y_value = arg_y->BuildValue()->cast<ValueTuplePtr>(); in InferImplBroadcastGradientArgs()
414 auto shpx_data = shpx_value->cast<ValueTuplePtr>()->value(); in InferImplTupleDiv()
415 auto div_shp_data = div_shp_value->cast<ValueTuplePtr>()->value(); in InferImplTupleDiv()
469 auto shpx_data = shpx_value->cast<ValueTuplePtr>()->value(); in InferImplShapeMul()
/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils.cc282 auto value_tuple = value->cast<ValueTuplePtr>(); in TensorValueToTensor()
301 size_t CountValueNum(const ValueTuplePtr &value_tuple) { in CountValueNum()
309 cnt += CountValueNum(value->cast<ValueTuplePtr>()); in CountValueNum()
Dconvert_utils.h79 size_t CountValueNum(const ValueTuplePtr &value_tuple);
/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/frontend/optimizer/irpass/
Ditem_tuple_or_list_eliminate.h165 tuple_ = GetValueNode<ValueTuplePtr>(vnode); in Visit()
189 ValueTuplePtr tuple_{nullptr};
248 auto tuple = GetValueNode<ValueTuplePtr>(vnode); in Visit()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h162 tensor::TensorPtr CreateTensorWithValueTuple(const ValueTuplePtr &value_tuple_ptr, const TypePtr &t…
165 tensor::TensorPtr CreateTupleTensor(const ValueTuplePtr &value_tuple);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconst_to_attr_strided_slice_grad.cc52 auto value_tuple = value->cast<ValueTuplePtr>(); in GetStridesValues()
Dconvert_const_input_to_tensor_input.cc41 tensor_ptr = CreateTupleTensor(value->cast<ValueTuplePtr>()); in CreateTensorInput()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc73 inputs_seq = node_inputs[1]->cast<ValueNodePtr>()->value()->cast<ValueTuplePtr>()->value(); in ExtractInputParameterByNode()
163 inputs_seq = node_inputs[1]->cast<ValueNodePtr>()->value()->cast<ValueTuplePtr>()->value(); in ExtractInputTypeLengthByNode()

123