|
|
|
@@ -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}}, |
|
|
|
|