| 911 | } |
| 912 | |
| 913 | mluOpStatus_t MLUOP_WIN_API mluOpInitTensorSetMemberDescriptor( |
| 914 | mluOpTensorSetDescriptor_t tensorSet, const int tensorSetDimNb, |
| 915 | const int *tensorIndex, mluOpTensorLayout_t layout, mluOpDataType_t dtype, |
| 916 | const int dimNb, const int *dimSize) { |
| 917 | PARAM_CHECK("[mluOpInitTensorSetMemberDescriptor]", |
| 918 | tensorSet->dim_num == tensorSetDimNb); |
| 919 | auto ts = tensorSet->getTensor(tensorIndex); |
| 920 | PARAM_CHECK("[mluOpInitTensorSetMemberDescriptor]", |
| 921 | MLUOP_STATUS_SUCCESS == |
| 922 | mluOpSetTensorDescriptor(ts, layout, dtype, dimNb, dimSize)); |
| 923 | return MLUOP_STATUS_SUCCESS; |
| 924 | } |
| 925 | |
| 926 | mluOpStatus_t MLUOP_WIN_API mluOpInitTensorSetMemberDescriptor_v2( |
| 927 | mluOpTensorSetDescriptor_t tensorSet, const int tensorSetDimNb, |