From 269e898e23cfb4acb337beb5385620ac5afd9df8 Mon Sep 17 00:00:00 2001 From: Kevin Cheng Date: Tue, 28 Sep 2021 15:48:27 -0700 Subject: Remove ReluN op. Signed-off-by: Kevin Cheng Change-Id: Iaac727159a84de1f83de549c3a22704096f46bf9 --- reference_model/src/ops/activation_funcs.cc | 28 ---------------------------- 1 file changed, 28 deletions(-) (limited to 'reference_model/src/ops/activation_funcs.cc') diff --git a/reference_model/src/ops/activation_funcs.cc b/reference_model/src/ops/activation_funcs.cc index 440f4e1..21677d5 100644 --- a/reference_model/src/ops/activation_funcs.cc +++ b/reference_model/src/ops/activation_funcs.cc @@ -50,31 +50,6 @@ int OpClamp::register_fcn() return 0; } -template -int OpReluN::register_fcn() -{ - - switch (Dtype) - { - case DType_FLOAT: - { - InEigenType N = (InEigenType)attribute->max_fp(); - this->fcn = [N](InEigenType a) -> OutEigenType { return a >= 0 ? (a <= N ? a : N) : 0; }; - } - break; - case DType_INT32: - { - InEigenType N = (InEigenType)attribute->max_int(); - this->fcn = [N](InEigenType a) -> OutEigenType { return a >= 0 ? (a <= N ? a : N) : 0; }; - } - break; - default: - ERROR_IF(true, "unsupported DType %s", EnumNamesDType()[Dtype]); - } - - return 0; -} - template int OpSigmoid::register_fcn() { @@ -110,9 +85,6 @@ DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpClamp, FLOAT); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpClamp, INT8); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpClamp, INT16); -DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpReluN, FLOAT); -DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpReluN, INT32); - DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpSigmoid, FLOAT); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpTanh, FLOAT); -- cgit v1.2.1