diff --git a/mindspore/core/abstract/primitive_infer_map.cc b/mindspore/core/abstract/primitive_infer_map.cc index 43ea9cf392..01b6f4530d 100644 --- a/mindspore/core/abstract/primitive_infer_map.cc +++ b/mindspore/core/abstract/primitive_infer_map.cc @@ -123,7 +123,6 @@ PrimitiveEvalImplMap &GetPrimitiveToEvalImplMap() { {prim::kPrimReluGrad, {InferImplReluGrad, true}}, {prim::kPrimConv2D, {InferImplConv2D, true}}, {prim::kPrimBiasAdd, {InferImplBiasAdd, true}}, - {prim::kPrimBiasAddGrad, {InferImplBiasAddGrad, true}}, {prim::kPrimRelu, {InferImplRelu, true}}, {prim::kPrimRelu6, {InferImplRelu, true}}, {prim::kPrimZerosLike, {InferImplZerosLike, true}}, @@ -186,6 +185,7 @@ PrimitiveEvalImplMap &GetPrimitiveToBackendEvalImplMap() { {prim::kPrimReduceAny, {InferImplReduceFunc, true}}, {prim::kPrimReduceMax, {InferImplReduceFunc, true}}, {prim::kPrimReduceMin, {InferImplReduceFunc, true}}, + {prim::kPrimBiasAddGrad, {InferImplBiasAddGrad, true}}, {prim::kPrimReduceScatter, {InferImplReduceScatter, true}}, {prim::kPrimCast, {InferImplCast, true}}, {prim::kPrimExpandDims, {InferImplExpandDims, true}}, diff --git a/tests/ut/cpp/pre_activate/common/restore_abs_input_in_backed_infer_test.cc b/tests/ut/cpp/pre_activate/common/restore_abs_input_in_backed_infer_test.cc index bfabc1998d..8cb97d3a5f 100644 --- a/tests/ut/cpp/pre_activate/common/restore_abs_input_in_backed_infer_test.cc +++ b/tests/ut/cpp/pre_activate/common/restore_abs_input_in_backed_infer_test.cc @@ -1,7 +1,5 @@ /** - * This is the C++ adaptation and derivative work of Myia (https://github.com/mila-iqia/myia/). - * - * Copyright 2019-2021 Huawei Technologies Co., Ltd + * Copyright 2021 Huawei Technologies Co., Ltd * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License.