| 949 | } |
| 950 | |
| 951 | mluOpStatus_t MLUOP_WIN_API mluOpGetTensorSetDescriptorSize( |
| 952 | mluOpTensorSetDescriptor_t tensorSet, int *sizeInBytes) { |
| 953 | PARAM_CHECK("[mluOpGetTensorSetDescriptorSize]", tensorSet != NULL); |
| 954 | int tensor_set_size = tensorSet->getSize(); |
| 955 | *sizeInBytes = tensor_set_size; |
| 956 | return MLUOP_STATUS_SUCCESS; |
| 957 | } |
| 958 | |
| 959 | mluOpStatus_t MLUOP_WIN_API mluOpGetTensorAndDataFromTensorSet( |
| 960 | mluOpTensorSetDescriptor_t tensorSet, const int tensorSetDimNb, |