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

fix_ios

Browse Source
pull/14607/head
yefeng 4 years ago
parent 5298b1d225
commit dbf02f0756
123 changed files with 1645 additions and 210 deletions
Show Stats Download Patch File Download Diff File

352
mindspore/lite/nnacl/infer/infer_register.c
View File

File diff suppressed because it is too large Load Diff

4
mindspore/lite/src/ops/populate/activation_grad_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/fp32_grad/activation_grad.h"
using mindspore::schema::PrimitiveType_ActivationGrad;
namespace mindspore {
namespace lite {
@ -34,7 +35,6 @@ OpParameter *PopulateActivationGradParameter(const void *prim) {
act_param->alpha_ = value->alpha();
return reinterpret_cast<OpParameter *>(act_param);
}
Registry ActivationGradParameterRegistry(schema::PrimitiveType_ActivationGrad, PopulateActivationGradParameter,
SCHEMA_CUR);
REG_POPULATE(PrimitiveType_ActivationGrad, PopulateActivationGradParameter, SCHEMA_CUR);
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/activation_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/fp32/activation_fp32.h"
using mindspore::schema::PrimitiveType_Activation;
namespace mindspore {
namespace lite {
@ -37,6 +38,6 @@ OpParameter *PopulateRelu6Parameter(const void *prim) {
}
} // namespace
Registry g_relu6ParameterRegistry(schema::PrimitiveType_Activation, PopulateRelu6Parameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Activation, PopulateRelu6Parameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/adam_populate.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Adam;
namespace mindspore {
namespace lite {
@ -29,6 +30,6 @@ OpParameter *PopulateAdamParameter(const void *prim) {
return param;
}
Registry AdamParameterRegistry(schema::PrimitiveType_Adam, PopulateAdamParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Adam, PopulateAdamParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/add_populate.cc
Unescape View File

@ -16,6 +16,7 @@
#include "src/ops/populate/populate_register.h"
#include "nnacl/arithmetic.h"
#include "src/ops/populate/arithmetic_populate.h"
using mindspore::schema::PrimitiveType_AddFusion;
namespace mindspore {
namespace lite {
@ -33,6 +34,6 @@ OpParameter *PopulateAddParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(param);
}
} // namespace
Registry g_addParameterRegistry(schema::PrimitiveType_AddFusion, PopulateAddParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_AddFusion, PopulateAddParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/adder_populate.cc
Unescape View File

@ -16,6 +16,7 @@
#include "src/common/log_adapter.h"
#include "nnacl/conv_parameter.h"
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_AdderFusion;
namespace mindspore {
namespace lite {
@ -57,6 +58,6 @@ OpParameter *PopulateAdderParameter(const void *prim) {
}
return reinterpret_cast<OpParameter *>(conv_param);
}
Registry g_AdderParameterRegistry(schema::PrimitiveType_AdderFusion, PopulateAdderParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_AdderFusion, PopulateAdderParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/addn_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/op_base.h"
using mindspore::schema::PrimitiveType_AddN;
namespace mindspore {
namespace lite {
@ -31,6 +32,6 @@ OpParameter *PopulateAddNParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(addn_param);
}
} // namespace
Registry g_addNParameterRegistry(schema::PrimitiveType_AddN, PopulateAddNParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_AddN, PopulateAddNParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/argmax_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/arg_min_max_parameter.h"
using mindspore::schema::PrimitiveType_ArgMaxFusion;
namespace mindspore {
namespace lite {
@ -38,6 +39,6 @@ OpParameter *PopulateArgMaxParameter(const void *prim) {
}
} // namespace
Registry g_argMaxParameterRegistry(schema::PrimitiveType_ArgMaxFusion, PopulateArgMaxParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_ArgMaxFusion, PopulateArgMaxParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/argmin_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/arg_min_max_parameter.h"
using mindspore::schema::PrimitiveType_ArgMinFusion;
namespace mindspore {
namespace lite {
@ -38,6 +39,6 @@ OpParameter *PopulateArgMinParameter(const void *prim) {
}
} // namespace
Registry g_argMinParameterRegistry(schema::PrimitiveType_ArgMinFusion, PopulateArgMinParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_ArgMinFusion, PopulateArgMinParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

51
mindspore/lite/src/ops/populate/arithmetic_populate.cc
Unescape View File

@ -15,6 +15,23 @@
*/
#include "src/ops/populate/arithmetic_populate.h"
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Equal;
using mindspore::schema::PrimitiveType_FloorDiv;
using mindspore::schema::PrimitiveType_FloorMod;
using mindspore::schema::PrimitiveType_Greater;
using mindspore::schema::PrimitiveType_GreaterEqual;
using mindspore::schema::PrimitiveType_Less;
using mindspore::schema::PrimitiveType_LessEqual;
using mindspore::schema::PrimitiveType_LogicalAnd;
using mindspore::schema::PrimitiveType_LogicalOr;
using mindspore::schema::PrimitiveType_Maximum;
using mindspore::schema::PrimitiveType_MaximumGrad;
using mindspore::schema::PrimitiveType_Minimum;
using mindspore::schema::PrimitiveType_MinimumGrad;
using mindspore::schema::PrimitiveType_Mod;
using mindspore::schema::PrimitiveType_NotEqual;
using mindspore::schema::PrimitiveType_RealDiv;
using mindspore::schema::PrimitiveType_SquaredDifference;
namespace mindspore {
namespace lite {
@ -42,22 +59,22 @@ OpParameter *PopulateArithmetic(const void *primitive) {
return reinterpret_cast<OpParameter *>(param);
}
Registry g_MinimunGradParameterRegistry(schema::PrimitiveType_MinimumGrad, PopulateArithmetic, SCHEMA_CUR);
Registry g_MaximunGradParameterRegistry(schema::PrimitiveType_MaximumGrad, PopulateArithmetic, SCHEMA_CUR);
Registry g_realDivParameterRegistry(schema::PrimitiveType_RealDiv, PopulateArithmetic, SCHEMA_CUR);
Registry g_logicalAndParameterRegistry(schema::PrimitiveType_LogicalAnd, PopulateArithmetic, SCHEMA_CUR);
Registry g_parameterRegistry(schema::PrimitiveType_LogicalOr, PopulateArithmetic, SCHEMA_CUR);
Registry g_equalParameterRegistry(schema::PrimitiveType_Equal, PopulateArithmetic, SCHEMA_CUR);
Registry g_notEqualParameterRegistry(schema::PrimitiveType_NotEqual, PopulateArithmetic, SCHEMA_CUR);
Registry g_essParameterRegistry(schema::PrimitiveType_Less, PopulateArithmetic, SCHEMA_CUR);
Registry g_lessEqualParameterRegistry(schema::PrimitiveType_LessEqual, PopulateArithmetic, SCHEMA_CUR);
Registry g_greaterParameterRegistry(schema::PrimitiveType_Greater, PopulateArithmetic, SCHEMA_CUR);
Registry g_greaterEqualParameterRegistry(schema::PrimitiveType_GreaterEqual, PopulateArithmetic, SCHEMA_CUR);
Registry g_maximumParameterRegistry(schema::PrimitiveType_Maximum, PopulateArithmetic, SCHEMA_CUR);
Registry g_minimumParameterRegistry(schema::PrimitiveType_Minimum, PopulateArithmetic, SCHEMA_CUR);
Registry g_floorDivParameterRegistry(schema::PrimitiveType_FloorDiv, PopulateArithmetic, SCHEMA_CUR);
Registry g_floorModParameterRegistry(schema::PrimitiveType_FloorMod, PopulateArithmetic, SCHEMA_CUR);
Registry g_modParameterRegistry(schema::PrimitiveType_Mod, PopulateArithmetic, SCHEMA_CUR);
Registry g_squaredDifferenceParameterRegistry(schema::PrimitiveType_SquaredDifference, PopulateArithmetic, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_MinimumGrad, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_MaximumGrad, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_RealDiv, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_LogicalAnd, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_LogicalOr, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Equal, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_NotEqual, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Less, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_LessEqual, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Greater, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_GreaterEqual, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Maximum, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Minimum, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_FloorDiv, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_FloorMod, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Mod, PopulateArithmetic, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_SquaredDifference, PopulateArithmetic, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

45
mindspore/lite/src/ops/populate/arithmetic_self_populate.cc
Unescape View File

@ -16,6 +16,21 @@
#include "src/common/log_adapter.h"
#include "nnacl/arithmetic_self_parameter.h"
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Abs;
using mindspore::schema::PrimitiveType_Ceil;
using mindspore::schema::PrimitiveType_Cos;
using mindspore::schema::PrimitiveType_Floor;
using mindspore::schema::PrimitiveType_Log;
using mindspore::schema::PrimitiveType_LogGrad;
using mindspore::schema::PrimitiveType_LogicalNot;
using mindspore::schema::PrimitiveType_Neg;
using mindspore::schema::PrimitiveType_NegGrad;
using mindspore::schema::PrimitiveType_Reciprocal;
using mindspore::schema::PrimitiveType_Round;
using mindspore::schema::PrimitiveType_Rsqrt;
using mindspore::schema::PrimitiveType_Sin;
using mindspore::schema::PrimitiveType_Sqrt;
using mindspore::schema::PrimitiveType_Square;
namespace mindspore {
namespace lite {
@ -32,20 +47,20 @@ OpParameter *PopulateArithmeticSelf(const void *prim) {
return reinterpret_cast<OpParameter *>(arithmetic_self_param);
}
Registry g_absParameterRegistry(schema::PrimitiveType_Abs, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_cosParameterRegistry(schema::PrimitiveType_Cos, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_sinParameterRegistry(schema::PrimitiveType_Sin, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_logParameterRegistry(schema::PrimitiveType_Log, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_negParameterRegistry(schema::PrimitiveType_Neg, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_negGradParameterRegistry(schema::PrimitiveType_NegGrad, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_logGradParameterRegistry(schema::PrimitiveType_LogGrad, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_sqrtParameterRegistry(schema::PrimitiveType_Sqrt, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_squareParameterRegistry(schema::PrimitiveType_Square, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_rsqrtParameterRegistry(schema::PrimitiveType_Rsqrt, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_logicalNotParameterRegistry(schema::PrimitiveType_LogicalNot, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_floorParameterRegistry(schema::PrimitiveType_Floor, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_ceilParameterRegistry(schema::PrimitiveType_Ceil, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_roundParameterRegistry(schema::PrimitiveType_Round, PopulateArithmeticSelf, SCHEMA_CUR);
Registry g_reciprocalParameterRegistry(schema::PrimitiveType_Reciprocal, PopulateArithmeticSelf, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Abs, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Cos, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Sin, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Log, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Neg, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_NegGrad, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_LogGrad, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Sqrt, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Square, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Rsqrt, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_LogicalNot, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Floor, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Ceil, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Round, PopulateArithmeticSelf, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Reciprocal, PopulateArithmeticSelf, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/assert_populate.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Assert;
namespace mindspore {
namespace lite {
@ -30,6 +31,6 @@ OpParameter *PopulateAssertParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(assert_parameter);
}
Registry AssertParameterRegistry(schema::PrimitiveType_Assert, PopulateAssertParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Assert, PopulateAssertParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/assign_add_populate.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_AssignAdd;
namespace mindspore {
namespace lite {
@ -29,6 +30,6 @@ OpParameter *PopulateAssignAddParameter(const void *prim) {
return param;
}
Registry AssignAddParameterRegistry(schema::PrimitiveType_AssignAdd, PopulateAssignAddParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_AssignAdd, PopulateAssignAddParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/assign_populate.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Assign;
namespace mindspore {
namespace lite {
@ -30,6 +31,6 @@ OpParameter *PopulateAssignParameter(const void *prim) {
return param;
}
Registry AssignParameterRegistry(schema::PrimitiveType_Assign, PopulateAssignParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Assign, PopulateAssignParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/audio_spectrogram_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/infer/audio_spectrogram_infer.h"
using mindspore::schema::PrimitiveType_AudioSpectrogram;
namespace mindspore {
namespace lite {
@ -36,7 +37,6 @@ OpParameter *PopulateAudioSpectrogramParameter(const void *prim) {
}
} // namespace
Registry g_audioSpectrogramParameterRegistry(schema::PrimitiveType_AudioSpectrogram, PopulateAudioSpectrogramParameter,
SCHEMA_CUR);
REG_POPULATE(PrimitiveType_AudioSpectrogram, PopulateAudioSpectrogramParameter, SCHEMA_CUR);
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/batch_norm_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/batchnorm_parameter.h"
using mindspore::schema::PrimitiveType_BatchNorm;
namespace mindspore {
namespace lite {
@ -35,7 +36,6 @@ OpParameter *PopulateBatchNorm(const void *prim) {
}
} // namespace
Registry g_batchNormParameterRegistry(schema::PrimitiveType_BatchNorm, PopulateBatchNorm, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BatchNorm, PopulateBatchNorm, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

6
mindspore/lite/src/ops/populate/batch_to_space_populate.cc
Unescape View File

@ -15,6 +15,8 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/batch_to_space.h"
using mindspore::schema::PrimitiveType_BatchToSpace;
using mindspore::schema::PrimitiveType_BatchToSpaceND;
namespace mindspore {
namespace lite {
@ -63,7 +65,7 @@ OpParameter *PopulateBatchToSpaceParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(batch_space_param);
}
} // namespace
Registry g_batchToSpaceRegistry(schema::PrimitiveType_BatchToSpace, PopulateBatchToSpaceParameter, SCHEMA_CUR);
Registry g_batchToSpaceNDRegistry(schema::PrimitiveType_BatchToSpaceND, PopulateBatchToSpaceParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BatchToSpace, PopulateBatchToSpaceParameter, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_BatchToSpaceND, PopulateBatchToSpaceParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/bias_add_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/arithmetic.h"
using mindspore::schema::PrimitiveType_BiasAdd;
namespace mindspore {
namespace lite {
@ -32,7 +33,6 @@ OpParameter *PopulateBiasAddParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(arithmetic_param);
}
} // namespace
Registry g_biasAddParameterRegistry(schema::PrimitiveType_BiasAdd, PopulateBiasAddParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BiasAdd, PopulateBiasAddParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/bias_grad_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/arithmetic.h"
using mindspore::schema::PrimitiveType_BiasAddGrad;
namespace mindspore {
namespace lite {
@ -33,7 +34,6 @@ OpParameter *PopulateBiasAddGradParameter(const void *prim) {
}
} // namespace
Registry g_populateBiasGradParameterParameterRegistry(schema::PrimitiveType_BiasAddGrad, PopulateBiasAddGradParameter,
SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BiasAddGrad, PopulateBiasAddGradParameter, SCHEMA_CUR);
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/binary_cross_entropy_grad_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/fp32_grad/binary_cross_entropy_grad.h"
using mindspore::schema::PrimitiveType_BinaryCrossEntropyGrad;
namespace mindspore {
namespace lite {
@ -35,7 +36,6 @@ OpParameter *PopulateBinaryCrossEntropyGradParameter(const void *prim) {
}
} // namespace
Registry g_binaryCrossEntropyGradParameterRegistry(schema::PrimitiveType_BinaryCrossEntropyGrad,
PopulateBinaryCrossEntropyGradParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BinaryCrossEntropyGrad, PopulateBinaryCrossEntropyGradParameter, SCHEMA_CUR);
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/binary_cross_entropy_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/fp32_grad/binary_cross_entropy.h"
using mindspore::schema::PrimitiveType_BinaryCrossEntropy;
namespace mindspore {
namespace lite {
@ -33,7 +34,6 @@ OpParameter *PopulateBinaryCrossEntropyParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(bce_param);
}
Registry BinaryCrossEntropyParameterRegistry(schema::PrimitiveType_BinaryCrossEntropy,
PopulateBinaryCrossEntropyParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BinaryCrossEntropy, PopulateBinaryCrossEntropyParameter, SCHEMA_CUR);
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/broadcast_to_populate.cc
Unescape View File

@ -15,6 +15,7 @@
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/fp32/broadcast_to_fp32.h"
using mindspore::schema::PrimitiveType_BroadcastTo;
namespace mindspore {
namespace lite {
@ -37,6 +38,6 @@ OpParameter *PopulateBroadcastToParameter(const void *prim) {
return reinterpret_cast<OpParameter *>(broadcast_param);
}
Registry BroadcastToParameterRegistry(schema::PrimitiveType_BroadcastTo, PopulateBroadcastToParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_BroadcastTo, PopulateBroadcastToParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

3
mindspore/lite/src/ops/populate/cast_populate.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Cast;
namespace mindspore {
namespace lite {
@ -31,6 +32,6 @@ OpParameter *PopulateCastParameter(const void *prim) {
}
} // namespace
Registry g_castParameterRegistry(schema::PrimitiveType_Cast, PopulateCastParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Cast, PopulateCastParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

4
mindspore/lite/src/ops/populate/clip_populate.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Clip;
namespace mindspore {
namespace lite {
@ -31,7 +32,6 @@ OpParameter *PopulateClipParameter(const void *prim) {
}
} // namespace
Registry g_clipParameterRegistry(schema::PrimitiveType_Clip, PopulateClipParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_Clip, PopulateClipParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

6
mindspore/lite/src/ops/populate/common_populate.cc
Unescape View File

@ -14,6 +14,8 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
using mindspore::schema::PrimitiveType_Depend;
using mindspore::schema::PrimitiveType_ZerosLike;
namespace mindspore {
namespace lite {
@ -31,7 +33,7 @@ OpParameter *PopulateCommonParameter(const void *prim) {
}
} // namespace
Registry g_zerosLikeParameterRegistry(schema::PrimitiveType_ZerosLike, PopulateCommonParameter, SCHEMA_CUR);
Registry g_dependParameterRegistry(schema::PrimitiveType_Depend, PopulateCommonParameter, SCHEMA_CUR);
REG_POPULATE(PrimitiveType_ZerosLike, PopulateCommonParameter, SCHEMA_CUR)
REG_POPULATE(PrimitiveType_Depend, PopulateCommonParameter, SCHEMA_CUR)
} // namespace lite
} // namespace mindspore

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

Write Preview
Loading…
Cancel
Save