|
|
|
@@ -91,8 +91,6 @@ AbstractBasePtr LocalResponseNormalizationInfer(const abstract::AnalysisEnginePt |
|
|
|
return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), |
|
|
|
InferShape(primitive, input_args)->shape()); |
|
|
|
} |
|
|
|
REGISTER_PRIMITIVE_EVAL_IMPL(LocalResponseNormalization, prim::kPrimLocalResponseNormalization, |
|
|
|
LocalResponseNormalizationInfer); |
|
|
|
REGISTER_PRIMITIVE_C(kNameLocalResponseNormalization, LocalResponseNormalization); |
|
|
|
} // namespace ops |
|
|
|
} // namespace mindspore |