已合并
fix infershape for clip_by_value_v2, addcdiv, addcmul #844
ego创建于 1月22日
fix infershape for clip_by_value_v2, addcdiv, addcmul #844
已合并
从已删除 :master合入到cann/ops-mathmaster
共 3 个文件变更+18-7
| @@ -20,6 +20,10 @@ | |||
| 20 | using namespace ge; | 20 | using namespace ge; |
| 21 | 21 | ||
| 22 | namespace ops { | 22 | namespace ops { |
| 23 | -IMPL_OP_INFERSHAPE(ClipByValueV2).InferShape(Ops::Base::InferShape4Broadcast); | 23 | +static ge::graphStatus InferShape4ClipByValueV2(gert::InferShapeContext* context) |
| 24 | - | 24 | +{ |
| 25 | + const size_t INPUT_NUM_THREE = 3; | ||
| 26 | + return Ops::Base::InferShape4Broadcast(context, INPUT_NUM_THREE); | ||
| 27 | +} | ||
| 28 | +IMPL_OP_INFERSHAPE(ClipByValueV2).InferShape(InferShape4ClipByValueV2); | ||
| 25 | } // namespace ops | 29 | } // namespace ops |
| @@ -17,6 +17,10 @@ | |||
| 17 | // infershape | 17 | // infershape |
| 18 | using namespace ge; | 18 | using namespace ge; |
| 19 | namespace ops { | 19 | namespace ops { |
| 20 | - | 20 | +static ge::graphStatus InferShape4Addcdiv(gert::InferShapeContext* context) |
| 21 | -IMPL_OP_INFERSHAPE(Addcdiv).InferShape(Ops::Base::InferShape4Broadcast); | 21 | +{ |
| 22 | + const size_t INPUT_NUM_THREE = 3; | ||
| 23 | + return Ops::Base::InferShape4Broadcast(context, INPUT_NUM_THREE); | ||
| 24 | +} | ||
| 25 | +IMPL_OP_INFERSHAPE(Addcdiv).InferShape(InferShape4Addcdiv); | ||
| 22 | } // namespace ops | 26 | } // namespace ops |
| @@ -17,7 +17,10 @@ | |||
| 17 | 17 | ||
| 18 | using namespace ge; | 18 | using namespace ge; |
| 19 | namespace ops { | 19 | namespace ops { |
| 20 | - | 20 | +static ge::graphStatus InferShape4Addcmul(gert::InferShapeContext* context) |
| 21 | -IMPL_OP_INFERSHAPE(Addcmul).InferShape(Ops::Base::InferShape4Broadcast); | 21 | +{ |
| 22 | - | 22 | + const size_t INPUT_NUM_THREE = 3; |
| 23 | + return Ops::Base::InferShape4Broadcast(context, INPUT_NUM_THREE); | ||
| 24 | +} | ||
| 25 | +IMPL_OP_INFERSHAPE(Addcmul).InferShape(InferShape4Addcmul); | ||
| 23 | } | 26 | } |