已合并
update: 更新文件 04.03_matrix_multiplication_operator_development_with_high_level_api.ipynb #536
chenmingyu创建于 8月5日
update: 更新文件 04.03_matrix_multiplication_operator_development_with_high_level_api.ipynb #536
已合并
共 1 个文件变更+2-2
| @@ -1057,7 +1057,7 @@ | |||
| 1057 | " CHECK_ACL(aclrtSetDevice(deviceId));\n", | 1057 | " CHECK_ACL(aclrtSetDevice(deviceId));\n", |
| 1058 | " CHECK_ACL(aclrtCreateStream(&stream));\n", | 1058 | " CHECK_ACL(aclrtCreateStream(&stream));\n", |
| 1059 | "\n", | 1059 | "\n", |
| 1060 | - " // 定义MatmulLeakyrelu的张量形状\n", | 1060 | + " // 定义MatmulCustom的张量形状\n", |
| 1061 | " const std::vector<int64_t> shape_a = {1024, 256}; // 输入a形状\n", | 1061 | " const std::vector<int64_t> shape_a = {1024, 256}; // 输入a形状\n", |
| 1062 | " const std::vector<int64_t> shape_b = {256, 640}; // 输入b形状\n", | 1062 | " const std::vector<int64_t> shape_b = {256, 640}; // 输入b形状\n", |
| 1063 | " const std::vector<int64_t> shape_bias = {640}; // 输入bias形状\n", | 1063 | " const std::vector<int64_t> shape_bias = {640}; // 输入bias形状\n", |
| @@ -1477,7 +1477,7 @@ | |||
| 1477 | " CHECK_ACL(aclrtSetDevice(deviceId));\n", | 1477 | " CHECK_ACL(aclrtSetDevice(deviceId));\n", |
| 1478 | " CHECK_ACL(aclrtCreateStream(&stream));\n", | 1478 | " CHECK_ACL(aclrtCreateStream(&stream));\n", |
| 1479 | "\n", | 1479 | "\n", |
| 1480 | - " // 定义MatmulAbs的张量形状\n", | 1480 | + " // 定义MatmulCustom的张量形状\n", |
| 1481 | " const std::vector<int64_t> shape_a = {1024, 512}; // 输入a形状\n", | 1481 | " const std::vector<int64_t> shape_a = {1024, 512}; // 输入a形状\n", |
| 1482 | " const std::vector<int64_t> shape_b = {512, 2048}; // 输入b形状\n", | 1482 | " const std::vector<int64_t> shape_b = {512, 2048}; // 输入b形状\n", |
| 1483 | " const std::vector<int64_t> shape_bias = {2048}; // 输入bias形状\n", | 1483 | " const std::vector<int64_t> shape_bias = {2048}; // 输入bias形状\n", |