Home
last modified time | relevance | path

Searched refs:BuildValue (Results 1 – 25 of 53) sorted by relevance

123

/third_party/mindspore/mindspore/core/ops/
Drange.cc66 MS_EXCEPTION_IF_NULL(input_args[kInputIndex0]->BuildValue()); in RangeInfer()
67 MS_EXCEPTION_IF_NULL(input_args[kInputIndex1]->BuildValue()); in RangeInfer()
68 MS_EXCEPTION_IF_NULL(input_args[kInputIndex2]->BuildValue()); in RangeInfer()
69 auto start_tensor = input_args[kInputIndex0]->BuildValue()->cast<tensor::TensorPtr>(); in RangeInfer()
70 auto limit_tensor = input_args[kInputIndex1]->BuildValue()->cast<tensor::TensorPtr>(); in RangeInfer()
71 auto delta_tensor = input_args[kInputIndex2]->BuildValue()->cast<tensor::TensorPtr>(); in RangeInfer()
Dfill.cc36 auto dtype_value = input_dtype->BuildValue(); in FillInfer()
43 auto out_shape = GetValue<std::vector<int64_t>>(input_args[kInputIndex1]->BuildValue()); in FillInfer()
46 auto x_value = input_args[kInputIndex2]->BuildValue(); in FillInfer()
Dwhere.cc39 auto num = input_args[kInputIndex0]->BuildValue()->cast<tensor::TensorPtr>()->ElementsNum(); in WhereInfer()
41 auto num1 = input_args[kInputIndex1]->BuildValue()->cast<tensor::TensorPtr>()->ElementsNum(); in WhereInfer()
43 auto num2 = input_args[kInputIndex2]->BuildValue()->cast<tensor::TensorPtr>()->ElementsNum(); in WhereInfer()
Dzeros.cc33 auto shape_value = input_args[0]->BuildValue(); in ZerosInferShape()
43 auto dtype_value = input_args[1]->BuildValue(); in ZerosInferType()
Dones.cc32 auto shape_value = input_args[0]->BuildValue(); in OnesInferShape()
41 auto dtype_value = input_args[1]->BuildValue(); in OnesInferType()
Dassert.cc46 auto condition_values = GetValue<std::vector<bool>>(input_args[0]->BuildValue()); in AssertInfer()
59 …auto condition_value = reinterpret_cast<bool *>(input_args[0]->BuildValue()->cast<tensor::TensorPt… in AssertInfer()
Dselect.cc96 auto cond_value = input_args[kCondIndex]->BuildValue(); in SelectInferValue()
97 auto x = input_args[kXIndex]->BuildValue(); in SelectInferValue()
98 auto y = input_args[kYIndex]->BuildValue(); in SelectInferValue()
Dgather.cc54 auto axis_value_ptr = axis->BuildValue(); in GatherInfer()
61 axis_val = GetValue<int64_t>(axis->BuildValue()); in GatherInfer()
Dexpand_dims.cc41 auto dim_val = GetValue<int64_t>(input_args[1]->BuildValue()); in ExpandDimsInfer()
Dcustom_normalize.cc25 auto base_value = input_args[0]->BuildValue(); in CustomNormalizeInferShape()
Dtopk.cc49 auto k_v = GetValue<int>(input_args[1]->BuildValue()); in TopKInfer()
Dscatter_nd.cc27 auto shape_value = input_args[kInputIndex2]->BuildValue(); in InferShape()
Dreduce_sum.cc106 input_args[1]->BuildValue()->isa<AnyValue>()) { in InferShape()
133 axis_ptr = input_args[1]->BuildValue(); in InferShape()
Dstrided_slice.cc121 auto temp_strides = stride_arg->cast<abstract::AbstractTuplePtr>()->BuildValue(); in CheckAndGetValidStrides()
202 auto temp_begin_v = tuple_begin_v->BuildValue(); in StridedSliceInferShape()
208 auto temp_end_v = tuple_end_v->BuildValue(); in StridedSliceInferShape()
Dskip_gram.cc29 auto infer_value = input_args[0]->BuildValue(); in InferShape()
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Ddata_test.cc43 ASSERT_EQ(1, s1.BuildValue()->cast<Int64ImmPtr>()->value()); in TEST_F()
46 ASSERT_TRUE(s1.BuildValue()->isa<AnyValue>()); in TEST_F()
47 ASSERT_TRUE(s1.BuildValue()->isa<AnyValue>()); in TEST_F()
56 auto tup = t1.BuildValue()->cast<ValueTuplePtr>()->value(); in TEST_F()
65 ValuePtr abs_f1_built = abs_f1->BuildValue(); in TEST_F()
70 ValuePtr abs_fg1_built = abs_fg1->BuildValue(); in TEST_F()
76 ValuePtr func_tuple_built = abs_func_tuple->BuildValue(); in TEST_F()
81 ValuePtr func_list_built = abs_func_list->BuildValue(); in TEST_F()
86 func_tuple_built = abs_func_tuple->BuildValue(); in TEST_F()
/third_party/mindspore/mindspore/core/abstract/
Dprim_statement.cc132 ValuePtr t = args_spec_list[1]->BuildValue(); in InferImplIs_()
137 ValuePtr x = args_spec_list[0]->BuildValue(); in InferImplIs_()
148 ValuePtr t = args_spec_list[1]->BuildValue(); in InferImplIsNot()
154 ValuePtr x = args_spec_list[0]->BuildValue(); in InferImplIsNot()
165 ValuePtr key_value = key->BuildValue(); in IsInDict()
197 ValuePtr v = args_spec_list[0]->BuildValue(); in InferImplIsConstant()
Dprim_structures.cc54 ValuePtr keyPtr = key->BuildValue(); in InferImplMakeDict()
72 ValuePtr keyPtr = key->BuildValue(); in InferImplMakeKwarg()
89 ValuePtr key_value = key->BuildValue(); in InferImplExtractKwarg()
114 ValuePtr scalar_value = args_spec_list[index]->cast<AbstractScalarPtr>()->BuildValue(); in InferImplMakeSlice()
128 auto build_value = arg->BuildValue(); in InferImplMakeSlice()
165 ValuePtr index_value = index->BuildValue(); in InferTupleOrListGetItem()
198 ValuePtr index_value = index->BuildValue(); in InferTupleOrListSetItem()
245 ValuePtr key_value = key->BuildValue(); in InferImplDictGetItem()
268 ValuePtr key_value = key->BuildValue(); in InferImplDictSetItem()
Dprim_others.cc177 auto dependant_value = dependant_abstract->BuildValue(); in InferImplDepend()
232 auto dense_shape_value = dense_shape->BuildValue(); in InferImplMakeRowTensor()
342 auto dense_shape_value = dense_shape->BuildValue()->cast<ValueTuplePtr>(); in InferImplMakeSparseTensor()
413 auto recv_size_value_ptr = recv_size->BuildValue(); in InferImplAllSwap()
519 attr = type_abs->BuildValue(); in InferImplCast()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc53 ValuePtr x_value = input_x->BuildValue(); in InferImplTupleOrListEqual()
54 ValuePtr y_value = input_y->BuildValue(); in InferImplTupleOrListEqual()
63 auto arg_value = args_spec_list[0]->BuildValue(); in CalcSlidePara()
72 auto arg_value = args_spec_list[1]->BuildValue(); in CalcSlidePara()
81 auto arg_value = args_spec_list[2]->BuildValue(); in CalcSlidePara()
261 auto arg_x_value = arg_x->BuildValue()->cast<ValueTuplePtr>(); in InferImplBroadcastGradientArgs()
264 auto arg_y_value = arg_y->BuildValue()->cast<ValueTuplePtr>(); in InferImplBroadcastGradientArgs()
358 auto x_shp_value = shape_x->BuildValue(); in InferImplReduceShape()
378 auto axis_value = axis->BuildValue(); in InferImplReduceShape()
399 auto div_shp_value = div_shp->BuildValue(); in InferImplTupleDiv()
[all …]
/third_party/mindspore/mindspore/core/ops/grad/
Davg_pool_grad.cc25 MS_EXCEPTION_IF_NULL(input_args[0]->BuildValue()); in AvgPoolGradInfer()
26 auto origin_input_shape = GetValue<std::vector<int64_t>>(input_args[0]->BuildValue()); in AvgPoolGradInfer()
Dmax_pool_grad.cc24 MS_EXCEPTION_IF_NULL(input_args[0]->BuildValue()); in MaxPoolGradInfer()
Dflatten_grad.cc34 auto out_shape = GetValue<std::vector<int64_t>>(input_shape->BuildValue()); in FlattenGradInfer()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprim.cc297 py::object BuildValue(const ValuePtr &value_ptr) { in BuildValue() function
343 MS_EXCEPTION_IF_NULL(arg_tuple->BuildValue()); in AbstractTupleToPython()
344 if (arg_tuple->BuildValue()->isa<AnyValue>()) { in AbstractTupleToPython()
389 MS_EXCEPTION_IF_NULL(arg_list->BuildValue()); in AbstractListToPython()
390 if (arg_list->BuildValue()->isa<AnyValue>()) { in AbstractListToPython()
422 (*dic)[ATTR_MIN_VALUE] = BuildValue(min_value); in ConvertAbstractTensorToPython()
423 (*dic)[ATTR_MAX_VALUE] = BuildValue(max_value); in ConvertAbstractTensorToPython()
427 (*dic)[ATTR_VALUE] = BuildValue(arg_tensor->BuildValue()); in ConvertAbstractTensorToPython()
439 MS_EXCEPTION_IF_NULL(args[0]->BuildValue()); in ConvertAbstractFunctionToPython()
440 auto value = args[0]->BuildValue()->cast<parse::ClassTypePtr>(); in ConvertAbstractFunctionToPython()
[all …]
/third_party/mindspore/mindspore/core/ops/fusion/
Dslice_fusion.cc39 auto begin_v = input_args[kInputIndex1]->BuildValue(); in SliceFusionInfer()
40 auto size_v = input_args[kInputIndex2]->BuildValue(); in SliceFusionInfer()

123