|
|
@@ -172,7 +172,7 @@ class OpUtils { |
|
|
|
|
|
|
|
/// |
|
|
|
/// @ingroup domi_omg |
|
|
|
/// @brief Convert the convolution¡®s weight data from [h, w, c, k] to [k, c, h, w] |
|
|
|
/// @brief Convert the convolution weight data from [h, w, c, k] to [k, c, h, w] |
|
|
|
/// @param [in] input Weight data in HWCK format |
|
|
|
/// @param [in] H value of H dimension |
|
|
|
/// @param [in] W value of W dimension |
|
|
@@ -183,7 +183,7 @@ class OpUtils { |
|
|
|
static void TransDataHWCK2KCHW(const void *input, int64_t H, int64_t W, int64_t C, int64_t K, void **output); |
|
|
|
/// |
|
|
|
/// @ingroup domi_omg |
|
|
|
/// @brief Converts the convolution¡®s weight data from [k, c, h, w] to [h, w, c, k]. |
|
|
|
/// @brief Converts the convolution weight data from [k, c, h, w] to [h, w, c, k]. |
|
|
|
/// @param [in] input Weight data in HWCK format |
|
|
|
/// @param [in] K value of K dimension |
|
|
|
/// @param [in] C value of C dimension |
|
|
|