已合并
fix infershape for clip_by_value_v2, addcdiv, addcmul #844
ego创建于 1月22日
fix infershape for clip_by_value_v2, addcdiv, addcmul #844
已合并
ego创建于 1月22日
已删除 :master合入到cann/ops-mathmaster
3 个文件变更+18-7
@@ -20,6 +20,10 @@
20using namespace ge;20using namespace ge;
21 21 
22namespace ops {22namespace 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 ops29} // namespace ops
@@ -17,6 +17,10 @@
17// infershape17// infershape
18using namespace ge;18using namespace ge;
19namespace ops {19namespace 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 ops26} // namespace ops
@@ -17,7 +17,10 @@
17 17 
18using namespace ge;18using namespace ge;
19namespace ops {19namespace 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}