| /third_party/mindspore/mindspore/core/ops/ |
| D | lerp.cc | 28 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
32 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
33 for (const auto &item : input_args) { in InferShape()
36 …auto start_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->B… in InferShape()
38 …auto end_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->Bui… in InferShape()
40 …auto weight_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->… in InferShape()
43 if (input_args[kInputIndex2]->isa<abstract::AbstractTensor>()) { in InferShape()
51 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
52 for (const auto &item : input_args) { in InferType()
57 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferType()
[all …]
|
| D | masked_fill.cc | 27 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
31 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, op_name); in InferShape()
32 …auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->B… in InferShape()
34 …auto mask_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->Bu… in InferShape()
36 …auto value_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->B… in InferShape()
39 if (input_args[kInputIndex2]->isa<abstract::AbstractTensor>()) { in InferShape()
51 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
52 for (const auto &item : input_args) { in InferType()
57 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, i… in InferType()
58 …(void)CheckAndConvertUtils::CheckTensorTypeValid("mask", input_args[1]->BuildType(), {kBool}, op_n… in InferType()
[all …]
|
| D | apply_adagrad_d_a.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
33 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
35 for (const auto &item : input_args) { in InferShape()
38 auto var_shape = input_args[0]->BuildShape(); in InferShape()
39 auto gradient_accumulator_shape = input_args[kInputIndex1]->BuildShape(); in InferShape()
40 auto gradient_squared_accumulator_shape = input_args[kInputIndex2]->BuildShape(); in InferShape()
41 …auto lr_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->BuildSha… in InferShape()
42 …auto l1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex5]->BuildSha… in InferShape()
43 …auto l2_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex6]->BuildSha… in InferShape()
45 CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex7]->BuildShape())[kShape]; in InferShape()
[all …]
|
| D | stack.cc | 22 …BasePtr StackInfer(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in StackInfer() argument
25 if (input_args.size() != 1) { in StackInfer()
26 MS_LOG(ERROR) << "Invalid output size:" << input_args.size(); in StackInfer()
28 if (input_args.size() < 1) { in StackInfer()
29 MS_LOG(ERROR) << "Invalid input size " << input_args.size(); in StackInfer()
31 for (const auto &item : input_args) { in StackInfer()
34 …auto input_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kS… in StackInfer()
35 for (size_t i = 1; i < input_args.size(); ++i) { in StackInfer()
36 …auto input_shape_tmp = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[i]->BuildShape()… in StackInfer()
47 …hape.insert(infer_shape.begin() + GetValue<int64_t>(primitive->GetAttr(kAxis)), input_args.size()); in StackInfer()
[all …]
|
| D | sparse_apply_r_m_s_prop.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
33 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, 6,… in InferShape()
34 for (const auto &item : input_args) { in InferShape()
37 auto var_shape_ptr = input_args[0]->BuildShape(); in InferShape()
38 auto ms_shape_ptr = input_args[1]->BuildShape(); in InferShape()
39 auto mom_shape_ptr = input_args[2]->BuildShape(); in InferShape()
43 …auto lr_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[3]->BuildShape())[kShap… in InferShape()
44 …auto grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[4]->BuildShape())[kSh… in InferShape()
45 …auto indices_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[5]->BuildShape())[… in InferShape()
72 TuplePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
[all …]
|
| D | binary_cross_entropy.cc | 33 const std::vector<AbstractBasePtr> &input_args) { in BinaryCrossEntroyInferShape() argument
36 …CheckAndConvertUtils::CheckInRange("binary_cross_entropy_infer", input_args.size(), kIncludeBoth, … in BinaryCrossEntroyInferShape()
37 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShap… in BinaryCrossEntroyInferShape()
38 …auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShap… in BinaryCrossEntroyInferShape()
39 …auto weight_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->Buil… in BinaryCrossEntroyInferShape()
52 …aryCrossEntroyInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in BinaryCrossEntroyInferType() argument
54 …(void)CheckAndConvertUtils::CheckInteger("binary_cross_entropy_infer", SizeToLong(input_args.size(… in BinaryCrossEntroyInferType()
56 for (const auto &item : input_args) { in BinaryCrossEntroyInferType()
61 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyInferType()
62 (void)types.emplace("y_shape", input_args[kInputIndex1]->BuildType()); in BinaryCrossEntroyInferType()
[all …]
|
| D | scatter_nd_update.cc | 26 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
27 …auto in_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShape… in InferShape()
28 …auto indices_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->Bui… in InferShape()
29 …auto update_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->Buil… in InferShape()
35 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
36 for (const auto &item : input_args) { in InferType()
42 auto input_x_type = input_args[kInputIndex0]->BuildType(); in InferType()
43 auto indices_type = input_args[kInputIndex1]->BuildType(); in InferType()
44 auto update_type = input_args[kInputIndex2]->BuildType(); in InferType()
48 return input_args[0]->BuildType(); in InferType()
[all …]
|
| D | sub.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
34 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
35 for (const auto &item : input_args) { in InferShape()
38 return BroadCastInferShape(prim_name, input_args); in InferShape()
41 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
42 for (const auto &item : input_args) { in InferType()
47 …(void)CheckAndConvertUtils::CheckInteger("infer", SizeToLong(input_args.size()), kGreaterEqual, in… in InferType()
49 (void)types.emplace("x", input_args[0]->BuildType()); in InferType()
50 (void)types.emplace("y", input_args[1]->BuildType()); in InferType()
56 const std::vector<AbstractBasePtr> &input_args) { in SubInfer() argument
[all …]
|
| D | mul.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
34 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
35 for (const auto &item : input_args) { in InferShape()
38 return BroadCastInferShape(prim_name, input_args); in InferShape()
41 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
42 for (const auto &item : input_args) { in InferType()
47 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kGreaterEq… in InferType()
50 (void)types.emplace("x", input_args[0]->BuildType()); in InferType()
51 (void)types.emplace("y", input_args[1]->BuildType()); in InferType()
57 const std::vector<AbstractBasePtr> &input_args) { in MulInfer() argument
[all …]
|
| D | real_div.cc | 29 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
33 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
34 for (const auto &item : input_args) { in InferShape()
37 return BroadCastInferShape(prim_name, input_args); in InferShape()
40 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
42 for (const auto &item : input_args) { in InferType()
47 …(void)CheckAndConvertUtils::CheckInteger("infer", SizeToLong(input_args.size()), kGreaterEqual, in… in InferType()
49 (void)types.emplace("x", input_args[0]->BuildType()); in InferType()
50 (void)types.emplace("y", input_args[1]->BuildType()); in InferType()
56 const std::vector<AbstractBasePtr> &input_args) { in RealDivInfer() argument
[all …]
|
| D | soft_shrink.cc | 31 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
34 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
36 for (const auto &item : input_args) { in InferShape()
39 …auto in_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShapeTrack())[kS… in InferShape()
42 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
45 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferType()
47 …if (std::any_of(input_args.begin(), input_args.end(), [](const AbstractBasePtr &a) { return a == n… in InferType()
51 …return CheckAndConvertUtils::CheckTensorTypeValid("input_x", input_args[0]->BuildType(), valid_typ… in InferType()
56 const std::vector<AbstractBasePtr> &input_args) { in SoftShrinkInfer() argument
57 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in SoftShrinkInfer()
[all …]
|
| D | scatter_nd.cc | 26 abstract::ShapePtr InferShape(const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
27 auto shape_value = input_args[kInputIndex2]->BuildValue(); in InferShape()
32 …auto indices_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->Bui… in InferShape()
33 …auto update_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->Buil… in InferShape()
39 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
40 for (const auto &item : input_args) { in InferType()
45 auto indices_type = input_args[0]->BuildType(); in InferType()
46 auto update_type = input_args[1]->BuildType(); in InferType()
49 return input_args[1]->BuildType(); in InferType()
54 const std::vector<AbstractBasePtr> &input_args) { in ScatterNdInfer() argument
[all …]
|
| D | select.cc | 44 …r SelectInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in SelectInferShape() argument
46 …auto cond_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kCondIndex]->BuildSha… in SelectInferShape()
47 auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kXIndex]->BuildShape()); in SelectInferShape()
48 auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kYIndex]->BuildShape()); in SelectInferShape()
53 if (CheckAndConvertUtils::HasDynamicShapeInput(input_args)) { in SelectInferShape()
62 MS_LOG(ERROR) << " cond shape :" << input_args[kCondIndex]->BuildShape()->ToString(); in SelectInferShape()
63 MS_LOG(ERROR) << " x shape :" << input_args[kXIndex]->BuildShape()->ToString(); in SelectInferShape()
64 MS_LOG(ERROR) << " y shape :" << input_args[kYIndex]->BuildShape()->ToString(); in SelectInferShape()
67 return input_args[1]->BuildShape(); in SelectInferShape()
70 TypePtr SelectInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in SelectInferType() argument
[all …]
|
| D | addn.cc | 28 …Ptr AddNInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in AddNInferShape() argument
30 …if (!input_args[0]->isa<abstract::AbstractTuple>() && !input_args[0]->isa<abstract::AbstractList>(… in AddNInferShape()
33 auto elements = input_args[0]->isa<abstract::AbstractTuple>() in AddNInferShape()
34 ? input_args[0]->cast<abstract::AbstractTuplePtr>()->elements() in AddNInferShape()
35 : input_args[0]->cast<abstract::AbstractListPtr>()->elements(); in AddNInferShape()
65 TypePtr AddNInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in AddNInferType() argument
68 …if (!input_args[0]->isa<abstract::AbstractTuple>() && !input_args[0]->isa<abstract::AbstractList>(… in AddNInferType()
71 auto elements = input_args[0]->isa<abstract::AbstractTuple>() in AddNInferType()
72 ? input_args[0]->cast<abstract::AbstractTuplePtr>()->elements() in AddNInferType()
73 : input_args[0]->cast<abstract::AbstractListPtr>()->elements(); in AddNInferType()
[all …]
|
| D | index_add.cc | 25 …IndexAddInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in IndexAddInferShape() argument
29 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, i… in IndexAddInferShape()
31 for (const auto &item : input_args) { in IndexAddInferShape()
34 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShap… in IndexAddInferShape()
35 …auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShap… in IndexAddInferShape()
41 …auto idx_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kSha… in IndexAddInferShape()
59 TypePtr IndexAddInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args)… in IndexAddInferType() argument
62 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in IndexAddInferType()
63 for (const auto &item : input_args) { in IndexAddInferType()
68 auto var_type = input_args[kInputIndex0]->BuildType(); in IndexAddInferType()
[all …]
|
| D | relu.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
33 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kGreaterE… in InferShape()
34 (void)CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferShape()
35 auto x = input_args[0]->BuildShape(); in InferShape()
41 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
44 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, 1… in InferType()
45 MS_EXCEPTION_IF_NULL(input_args[0]); in InferType()
46 auto x_type = input_args[0]->BuildType(); in InferType()
52 const std::vector<AbstractBasePtr> &input_args) { in ReLUInfer() argument
53 auto type = InferType(primitive, input_args); in ReLUInfer()
[all …]
|
| D | minimum.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
33 return BroadCastInferShape(op_name, input_args); in InferShape()
36 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
40 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferType()
41 …if (std::any_of(input_args.begin(), input_args.end(), [](const AbstractBasePtr &a) { return a == n… in InferType()
45 (void)types.emplace("x", input_args[0]->BuildType()); in InferType()
46 (void)types.emplace("y", input_args[1]->BuildType()); in InferType()
52 const std::vector<AbstractBasePtr> &input_args) { in MinimumInfer() argument
53 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in MinimumInfer()
54 InferShape(primitive, input_args)); in MinimumInfer()
|
| D | ctcloss.cc | 31 void CheckCTCLossInputs(const std::vector<AbstractBasePtr> &input_args, const std::string &op_name)… in CheckCTCLossInputs() argument
33 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kGreaterE… in CheckCTCLossInputs()
36 auto inputs = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 0); in CheckCTCLossInputs()
37 …labels_indices = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 1); in CheckCTCLossInputs()
38 … labels_values = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 2); in CheckCTCLossInputs()
39 …equence_length = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(op_name, input_args, 3); in CheckCTCLossInputs()
69 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
71 CheckCTCLossInputs(input_args, op_name); in InferShape()
72 auto input_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape()
93 TuplePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
[all …]
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | cdist_grad.cc | 26 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
29 for (const auto &item : input_args) { in InferShape()
32 …auto grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kSh… in InferShape()
33 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[kShape… in InferShape()
34 …auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[2]->BuildShape())[kShape… in InferShape()
35 …auto cdist_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[3]->BuildShape())[kS… in InferShape()
47 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
48 for (const auto &item : input_args) { in InferType()
53 (void)types.emplace("grad", input_args[0]->BuildType()); in InferType()
54 (void)types.emplace("input_x", input_args[1]->BuildType()); in InferType()
[all …]
|
| D | relu_grad.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
34 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
35 for (const auto &item : input_args) { in InferShape()
38 auto dout = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferShape()
39 auto out = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 1); in InferShape()
41 auto x = input_args[0]->BuildShape(); in InferShape()
48 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
52 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferType()
53 MS_EXCEPTION_IF_NULL(input_args[0]); in InferType()
54 auto dout = CheckAndConvertUtils::CheckArgs<abstract::AbstractTensor>(prim_name, input_args, 0); in InferType()
[all …]
|
| D | binary_cross_entropy_grad.cc | 27 const std::vector<AbstractBasePtr> &input_args) { in BinaryCrossEntroyGradInferShape() argument
30 …auto x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShap… in BinaryCrossEntroyGradInferShape()
31 …auto y_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShap… in BinaryCrossEntroyGradInferShape()
32 …auto weight_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->Buil… in BinaryCrossEntroyGradInferShape()
40 …rossEntroyGradInferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in BinaryCrossEntroyGradInferType() argument
43 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyGradInferType()
44 (void)types.emplace("y_shape", input_args[kInputIndex1]->BuildType()); in BinaryCrossEntroyGradInferType()
46 if (input_args[kInputIndex3]->BuildType() != nullptr) { in BinaryCrossEntroyGradInferType()
47 (void)types.emplace("x_shape", input_args[kInputIndex0]->BuildType()); in BinaryCrossEntroyGradInferType()
48 (void)types.emplace("weight_shape", input_args[kInputIndex2]->BuildType()); in BinaryCrossEntroyGradInferType()
[all …]
|
| D | soft_margin_loss_grad.cc | 28 const std::vector<AbstractBasePtr> &input_args) { in SoftMarginLossGradInferShape() argument
31 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, k… in SoftMarginLossGradInferShape()
32 …auto predict = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildShap… in SoftMarginLossGradInferShape()
33 …auto label = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->BuildShape(… in SoftMarginLossGradInferShape()
34 …auto dout = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildShape()… in SoftMarginLossGradInferShape()
42 …nLossGradInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in SoftMarginLossGradInferType() argument
44 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, k… in SoftMarginLossGradInferType()
47 (void)types.emplace("logits", input_args[kInputIndex0]->BuildType()); in SoftMarginLossGradInferType()
48 (void)types.emplace("labels", input_args[kInputIndex1]->BuildType()); in SoftMarginLossGradInferType()
49 (void)types.emplace("dout", input_args[kInputIndex2]->BuildType()); in SoftMarginLossGradInferType()
[all …]
|
| D | avg_pool_3d_grad.cc | 30 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
34 …(void)CheckAndConvertUtils::CheckInteger("input size", SizeToLong(input_args.size()), kEqual, inpu… in InferShape()
35 for (const auto &item : input_args) { in InferShape()
38 …auto grad_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShapeTrack())[… in InferShape()
41 if (input_args[0]->isa<abstract::AbstractTuple>()) { // origin_size is tuple in InferShape()
42 origin_input_size = GetValue<std::vector<int64_t>>(input_args[0]->BuildValue()); in InferShape()
49 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
53 …(void)CheckAndConvertUtils::CheckInteger("input size", SizeToLong(input_args.size()), kEqual, inpu… in InferType()
54 for (const auto &item : input_args) { in InferType()
57 auto grad_dtype = input_args[1]->BuildType(); in InferType()
[all …]
|
| D | hsigmoid_grad.cc | 33 …hapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferShape() argument
36 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
38 for (const auto &item : input_args) { in InferShape()
42 …auto grads_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kS… in InferShape()
43 …auto input_x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[… in InferShape()
48 TypePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
51 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, prim->name()); in InferType()
54 (void)types.emplace("grads", input_args[0]->BuildType()); in InferType()
55 (void)types.emplace("input_x", input_args[1]->BuildType()); in InferType()
61 const std::vector<AbstractBasePtr> &input_args) { in HSigmoidGradInfer() argument
[all …]
|
| /third_party/vk-gl-cts/android/cts/master/vk-master-2021-03-01/ |
| D | spirv-assembly.txt | 1257 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_add_var_flush_to_z…
1258 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_add_denorm_flush_t…
1259 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_add_inf_flush_to_z…
1260 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_sub_var_flush_to_z…
1261 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_sub_denorm_flush_t…
1262 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_sub_inf_flush_to_z…
1263 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_mul_var_flush_to_z…
1264 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_mul_denorm_flush_t…
1265 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_div_var_flush_to_z…
1266 dEQP-VK.spirv_assembly.instruction.compute.float_controls.fp16.input_args.denorm_div_inf_flush_to_z…
[all …]
|