Browse Source

fix bug of biasadd grad's infer shape

tags/v1.2.0-rc1
LianLiguang 4 years ago
parent
commit
d34074d1ac
2 changed files with 2 additions and 4 deletions
  1. +1
    -1
      mindspore/core/abstract/primitive_infer_map.cc
  2. +1
    -3
      tests/ut/cpp/pre_activate/common/restore_abs_input_in_backed_infer_test.cc

+ 1
- 1
mindspore/core/abstract/primitive_infer_map.cc View File

@@ -123,7 +123,6 @@ PrimitiveEvalImplMap &GetPrimitiveToEvalImplMap() {
{prim::kPrimReluGrad, {InferImplReluGrad, true}}, {prim::kPrimReluGrad, {InferImplReluGrad, true}},
{prim::kPrimConv2D, {InferImplConv2D, true}}, {prim::kPrimConv2D, {InferImplConv2D, true}},
{prim::kPrimBiasAdd, {InferImplBiasAdd, true}}, {prim::kPrimBiasAdd, {InferImplBiasAdd, true}},
{prim::kPrimBiasAddGrad, {InferImplBiasAddGrad, true}},
{prim::kPrimRelu, {InferImplRelu, true}}, {prim::kPrimRelu, {InferImplRelu, true}},
{prim::kPrimRelu6, {InferImplRelu, true}}, {prim::kPrimRelu6, {InferImplRelu, true}},
{prim::kPrimZerosLike, {InferImplZerosLike, true}}, {prim::kPrimZerosLike, {InferImplZerosLike, true}},
@@ -186,6 +185,7 @@ PrimitiveEvalImplMap &GetPrimitiveToBackendEvalImplMap() {
{prim::kPrimReduceAny, {InferImplReduceFunc, true}}, {prim::kPrimReduceAny, {InferImplReduceFunc, true}},
{prim::kPrimReduceMax, {InferImplReduceFunc, true}}, {prim::kPrimReduceMax, {InferImplReduceFunc, true}},
{prim::kPrimReduceMin, {InferImplReduceFunc, true}}, {prim::kPrimReduceMin, {InferImplReduceFunc, true}},
{prim::kPrimBiasAddGrad, {InferImplBiasAddGrad, true}},
{prim::kPrimReduceScatter, {InferImplReduceScatter, true}}, {prim::kPrimReduceScatter, {InferImplReduceScatter, true}},
{prim::kPrimCast, {InferImplCast, true}}, {prim::kPrimCast, {InferImplCast, true}},
{prim::kPrimExpandDims, {InferImplExpandDims, true}}, {prim::kPrimExpandDims, {InferImplExpandDims, true}},


+ 1
- 3
tests/ut/cpp/pre_activate/common/restore_abs_input_in_backed_infer_test.cc View File

@@ -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"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.


Loading…
Cancel
Save