已合并
Uniform/NormalV3 支持0维输入 #2515
梅国晗954517创建于 4月28日
Uniform/NormalV3 支持0维输入 #2515
已合并
共 6 个文件变更+6-6
| @@ -48,7 +48,7 @@ OpTilingConfig StatelessRandomNormalV3Tiling::BuildOpConfig() | |||
| 48 | {4, {{ge::DT_FLOAT}, -1, {}, nullptr}}, // stdev (scalar or tensor) | 48 | {4, {{ge::DT_FLOAT}, -1, {}, nullptr}}, // stdev (scalar or tensor) |
| 49 | }; | 49 | }; |
| 50 | config.outputCheckRules = { | 50 | config.outputCheckRules = { |
| 51 | - {0, {{ge::DT_FLOAT, ge::DT_BF16, ge::DT_FLOAT16}, -1, {1,2,3,4,5,6,7,8}, nullptr}} | 51 | + {0, {{ge::DT_FLOAT, ge::DT_BF16, ge::DT_FLOAT16}, -1, {0,1,2,3,4,5,6,7,8}, nullptr}} |
| 52 | }; // y | 52 | }; // y |
| 53 | config.getOutputSize = [](gert::TilingContext* ctx, int64_t& shapeSize) -> ge::graphStatus { | 53 | config.getOutputSize = [](gert::TilingContext* ctx, int64_t& shapeSize) -> ge::graphStatus { |
| 54 | auto outputShape = ctx->GetOutputShape(OUTPUT_IDX_Y); | 54 | auto outputShape = ctx->GetOutputShape(OUTPUT_IDX_Y); |
Mrandom/stateless_random_uniform_v3/op_host/arch35/stateless_random_uniform_v3_tiling_arch35.cpp+1-1
| @@ -48,7 +48,7 @@ OpTilingConfig StatelessRandomUniformV3Tiling::BuildOpConfig() | |||
| 48 | }; | 48 | }; |
| 49 | 49 | ||
| 50 | config.outputCheckRules = { | 50 | config.outputCheckRules = { |
| 51 | - {0, {{ge::DT_FLOAT, ge::DT_BF16, ge::DT_FLOAT16}, -1, {1,2,3,4,5,6,7,8}, nullptr}} // y | 51 | + {0, {{ge::DT_FLOAT, ge::DT_BF16, ge::DT_FLOAT16}, -1, {0,1,2,3,4,5,6,7,8}, nullptr}} // y |
| 52 | }; | 52 | }; |
| 53 | 53 | ||
| 54 | // 获取 outputSize：输出 y 的 shapeSize | 54 | // 获取 outputSize：输出 y 的 shapeSize |