已合并
【bugfix】math仓conversion和random支持非连续 #1817
majiajian_hw创建于 3月23日
【bugfix】math仓conversion和random支持非连续 #1817
已合并
majiajian_hw创建于 3月23日
共 2 个文件变更+8-4
@@ -29,7 +29,8 @@ public:
29 ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})29 ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
30 .UnknownShapeFormat(30 .UnknownShapeFormat(
31 {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND,31 {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND,
32- ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND});32+ ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
33+ .AutoContiguous();
33 this->Input("unique_indices")34 this->Input("unique_indices")
34 .ParamType(REQUIRED)35 .ParamType(REQUIRED)
35 .DataType(36 .DataType(
@@ -40,7 +41,8 @@ public:
40 ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})41 ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
41 .UnknownShapeFormat(42 .UnknownShapeFormat(
42 {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND,43 {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND,
43- ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND});44+ ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
45+ .AutoContiguous();
44 this->Input("indices")46 this->Input("indices")
45 .ParamType(REQUIRED)47 .ParamType(REQUIRED)
46 .DataType(48 .DataType(
@@ -63,7 +65,8 @@ public:
63 ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})65 ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
64 .UnknownShapeFormat(66 .UnknownShapeFormat(
65 {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND,67 {ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND,
66- ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND});68+ ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
69+ .AutoContiguous();
67 this->Output("new_indices")70 this->Output("new_indices")
68 .ParamType(REQUIRED)71 .ParamType(REQUIRED)
69 .DataType(72 .DataType(
@@ -24,7 +24,8 @@ public:
24 .ParamType(REQUIRED)24 .ParamType(REQUIRED)
25 .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT})25 .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT})
26 .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})26 .Format({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
27- .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND});27+ .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND, ge::FORMAT_ND})
28+ .AutoContiguous();
28 this->Output("self")29 this->Output("self")
29 .ParamType(REQUIRED)30 .ParamType(REQUIRED)
30 .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT})31 .DataType({ge::DT_BF16, ge::DT_FLOAT16, ge::DT_FLOAT})