m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!12634 remove cpp infer reg

Browse Source 
From: @lianliguang
Reviewed-by: 
Signed-off-by:
pull/12634/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent 674bca3fe4 2fa4997674
commit ac5b59c962
158 changed files with 4 additions and 222 deletions
Show Stats Download Patch File Download Diff File

1
mindspore/core/ops/abs.cc
Unescape View File

@ -56,7 +56,6 @@ AbstractBasePtr AbsInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(Abs, prim::kPrimAbs, AbsInfer);
REGISTER_PRIMITIVE_C(kNameAbs, Abs);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/adam.cc
Unescape View File

@ -81,8 +81,6 @@ AbstractBasePtr AdamInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt
const std::vector<AbstractBasePtr> &input_args) {
return std::make_shared<abstract::AbstractTensor>(AdamInfer(primitive, input_args));
}
REGISTER_PRIMITIVE_EVAL_IMPL(Adam, prim::kPrimAdam, AdamInfer);
REGISTER_PRIMITIVE_C(kNameAdam, Adam);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/add.cc
Unescape View File

@ -50,7 +50,6 @@ AbstractBasePtr AddInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(Add, prim::kPrimAdd, AddInfer);
REGISTER_PRIMITIVE_C(kNameAdd, Add);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/addn.cc
Unescape View File

@ -63,8 +63,6 @@ AbstractBasePtr AddNInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt
return std::make_shared<abstract::AbstractTensor>(TypeIdToType(infer_type),
std::make_shared<abstract::Shape>(element0_shape));
}
REGISTER_PRIMITIVE_EVAL_IMPL(AddN, prim::kPrimAddN, AddNInfer);
REGISTER_PRIMITIVE_C(kNameAddN, AddN);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/apply_momentum.cc
Unescape View File

@ -80,7 +80,6 @@ AbstractBasePtr ApplyMomentumInfer(const abstract::AnalysisEnginePtr &, const Pr
auto element = g_type_tensor->element();
return std::make_shared<abstract::AbstractTensor>(element, v_shape);
}
REGISTER_PRIMITIVE_EVAL_IMPL(ApplyMomentum, prim::kPrimApplyMomentum, ApplyMomentumInfer);
REGISTER_PRIMITIVE_C(kNameApplyMomentum, ApplyMomentum);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/arg_max.cc
Unescape View File

@ -67,7 +67,6 @@ AbstractBasePtr ArgMaxInfer(const abstract::AnalysisEnginePtr &, const Primitive
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(ArgMax, prim::kPrimArgMax, ArgMaxInfer);
REGISTER_PRIMITIVE_C(kNameArgMax, ArgMax);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/arg_min.cc
Unescape View File

@ -67,7 +67,6 @@ AbstractBasePtr ArgMinInfer(const abstract::AnalysisEnginePtr &, const Primitive
return std::make_shared<abstract::AbstractTensor>(x_dtype, std::make_shared<abstract::Shape>(out_shape));
}
REGISTER_PRIMITIVE_EVAL_IMPL(ArgMin, prim::kPrimArgMin, ArgMinInfer);
REGISTER_PRIMITIVE_C(kNameArgMin, ArgMin);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/asin.cc
Unescape View File

@ -46,7 +46,6 @@ AbstractBasePtr AsinInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt
return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(Asin, prim::kPrimAsin, AsinInfer);
REGISTER_PRIMITIVE_C(kNameAsin, Asin);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/assert.cc
Unescape View File

@ -71,8 +71,6 @@ AbstractBasePtr AssertInfer(const abstract::AnalysisEnginePtr &, const Primitive
}
return std::make_shared<abstract::AbstractTensor>(TypeIdToType(kNumberTypeInt32), output_shape);
}
REGISTER_PRIMITIVE_EVAL_IMPL(Assert, prim::kPrimAssert, AssertInfer);
REGISTER_PRIMITIVE_C(kNameAssert, Assert);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/assign_add.cc
Unescape View File

@ -47,7 +47,6 @@ AbstractBasePtr AssignAddInfer(const abstract::AnalysisEnginePtr &, const Primit
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(AssignAdd, prim::kPrimAssignAdd, AssignAddInfer);
REGISTER_PRIMITIVE_C(kNameAssignAdd, AssignAdd);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/atan.cc
Unescape View File

@ -44,7 +44,6 @@ AbstractBasePtr AtanInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt
return std::make_shared<abstract::AbstractTensor>(infer_type, infer_shape->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(Atan, prim::kPrimAtan, AtanInfer);
REGISTER_PRIMITIVE_C(kNameAtan, Atan);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/audio_spectrogram.cc
Unescape View File

@ -118,8 +118,6 @@ AbstractBasePtr AudioSpectrogramInfer(const abstract::AnalysisEnginePtr &, const
return std::make_shared<abstract::AbstractTensor>(AudioSpectrogramInferType(primitive, input_args),
AudioSpectrogramInferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(AudioSpectrogram, prim::kPrimAudioSpectrogram, AudioSpectrogramInfer);
REGISTER_PRIMITIVE_C(kNameAudioSpectrogram, AudioSpectrogram);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/avg_pool.cc
Unescape View File

@ -146,7 +146,6 @@ AbstractBasePtr AvgPoolInfer(const abstract::AnalysisEnginePtr &, const Primitiv
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(AvgPool, prim::kPrimAvgPool, AvgPoolInfer);
REGISTER_PRIMITIVE_C(kNameAvgPool, AvgPool);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/batch_norm.cc
Unescape View File

@ -134,7 +134,6 @@ AbstractBasePtr BatchNormInfer(const abstract::AnalysisEnginePtr &, const Primit
AbstractBasePtrList output = {output0, output1, output2, output3, output3};
return std::make_shared<abstract::AbstractTuple>(output);
}
REGISTER_PRIMITIVE_EVAL_IMPL(BatchNorm, prim::kPrimBatchNorm, BatchNormInfer);
REGISTER_PRIMITIVE_C(kNameBatchNorm, BatchNorm);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/batch_norm_fold.cc
Unescape View File

@ -109,8 +109,6 @@ AbstractBasePtr BatchNormFoldInfer(const abstract::AnalysisEnginePtr &, const Pr
AbstractBasePtrList output1 = {output, output, output, output};
return std::make_shared<abstract::AbstractTuple>(output1);
}
REGISTER_PRIMITIVE_EVAL_IMPL(BatchNormFold, prim::kPrimBatchNormFold, BatchNormFoldInfer);
REGISTER_PRIMITIVE_C(kNameBatchNormFold, BatchNormFold);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/batch_to_space.cc
Unescape View File

@ -75,7 +75,6 @@ AbstractBasePtr BatchToSpaceInfer(const abstract::AnalysisEnginePtr &, const Pri
ret->set_shape(std::make_shared<abstract::Shape>(out_shape));
return ret;
}
REGISTER_PRIMITIVE_EVAL_IMPL(BatchToSpace, prim::kPrimBatchToSpace, BatchToSpaceInfer);
REGISTER_PRIMITIVE_C(kNameBatchToSpace, BatchToSpace);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/batch_to_space_nd.cc
Unescape View File

@ -103,7 +103,6 @@ AbstractBasePtr BatchToSpaceNDInfer(const abstract::AnalysisEnginePtr &, const P
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(BatchToSpaceND, prim::kPrimBatchToSpaceND, BatchToSpaceNDInfer);
REGISTER_PRIMITIVE_C(kNameBatchToSpaceND, BatchToSpaceND);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/bias_add.cc
Unescape View File

@ -74,7 +74,6 @@ AbstractBasePtr BiasAddInfer(const abstract::AnalysisEnginePtr &, const Primitiv
BiasAddInferShape(primitive, input_args));
}
// Add
REGISTER_PRIMITIVE_EVAL_IMPL(BiasAdd, prim::kPrimBiasAdd, BiasAddInfer);
REGISTER_PRIMITIVE_C(kNameBiasAdd, BiasAdd);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/binary_cross_entropy.cc
Unescape View File

@ -87,8 +87,6 @@ AbstractBasePtr BinaryCrossEntropyInfer(const abstract::AnalysisEnginePtr &, con
return std::make_shared<abstract::AbstractTensor>(BinaryCrossEntroyInferType(primitive, input_args),
BinaryCrossEntroyInferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(BinaryCrossEntropy, prim::kPrimBinaryCrossEntropy, BinaryCrossEntropyInfer);
REGISTER_PRIMITIVE_C(kNameBinaryCrossEntropy, BinaryCrossEntropy);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/broadcast.cc
Unescape View File

@ -64,7 +64,6 @@ AbstractBasePtr BroadcastInfer(const abstract::AnalysisEnginePtr &, const Primit
}
return std::make_shared<abstract::AbstractTensor>(x_type, in_shape);
}
REGISTER_PRIMITIVE_EVAL_IMPL(Broadcast, prim::kPrimBroadcast, BroadcastInfer);
REGISTER_PRIMITIVE_C(kNameBroadcast, Broadcast);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/broadcast_to.cc
Unescape View File

@ -81,8 +81,6 @@ AbstractBasePtr BroadcastToInfer(const abstract::AnalysisEnginePtr &, const Prim
return std::make_shared<abstract::AbstractTensor>(BroadcastToInferType(primitive, input_args),
BroadcastToInferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(BroadcastTo, prim::kPrimBroadcastTo, BroadcastToInfer);
REGISTER_PRIMITIVE_C(kNameBroadcastTo, BroadcastTo);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/ceil.cc
Unescape View File

@ -43,7 +43,6 @@ AbstractBasePtr CeilInfer(const abstract::AnalysisEnginePtr &, const PrimitivePt
MS_EXCEPTION_IF_NULL(data_type);
return std::make_shared<abstract::AbstractTensor>(data_type, x_shape);
}
REGISTER_PRIMITIVE_EVAL_IMPL(Ceil, prim::kPrimCeil, CeilInfer);
REGISTER_PRIMITIVE_C(kNameCeil, Ceil);
} // namespace ops
} // namespace mindspore

2
mindspore/core/ops/concat.cc
Unescape View File

@ -78,8 +78,6 @@ AbstractBasePtr ConcatInfer(const abstract::AnalysisEnginePtr &, const Primitive
return std::make_shared<abstract::AbstractTensor>(TypeIdToType(infer_type),
std::make_shared<abstract::Shape>(ret_shape));
}
REGISTER_PRIMITIVE_EVAL_IMPL(Concat, prim::kPrimConcat, ConcatInfer);
REGISTER_PRIMITIVE_C(kNameConcat, Concat);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/constant.cc
Unescape View File

@ -52,7 +52,6 @@ AbstractBasePtr ConstantInfer(const abstract::AnalysisEnginePtr &, const Primiti
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(Constant, prim::kPrimConstant, ConstantInfer);
REGISTER_PRIMITIVE_C(kNameConstant, Constant);
} // namespace ops
} // namespace mindspore

1
mindspore/core/ops/constant_of_shape.cc
Unescape View File

@ -61,7 +61,6 @@ AbstractBasePtr ConstantOfShapeInfer(const abstract::AnalysisEnginePtr &, const
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args),
InferShape(primitive, input_args)->shape());
}
REGISTER_PRIMITIVE_EVAL_IMPL(ConstantOfShape, prim::kPrimConstantOfShape, ConstantOfShapeInfer);
REGISTER_PRIMITIVE_C(kNameConstantOfShape, ConstantOfShape);
} // namespace ops
} // namespace mindspore

Some files were not shown because too many files have changed in this diff Show More

Write Preview
Loading…
Cancel
Save