colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename Op 'Permute' to 'Transpose' 

Signed-off-by: Jiang Bo <bo.jiang@verisilicon.com>
This commit is contained in:
Jiang Bo 2021-01-12 11:21:51 +08:00
parent 7972af0697
commit 90b7a6fc32
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
include/tim/vx/ops/permute.h → include/tim/vx/ops/transpose.h
View File

@ -21,17 +21,17 @@
* DEALINGS IN THE SOFTWARE.
*
*****************************************************************************/
#ifndef TIM_VX_OPS_PERMUTE_H_
#define TIM_VX_OPS_PERMUTE_H_
#ifndef TIM_VX_OPS_TRANSPOSE_H_
#define TIM_VX_OPS_TRANSPOSE_H_
#include "tim/vx/operation.h"
namespace tim {
namespace vx {
namespace ops {
class Permute : public Operation {
class Transpose : public Operation {
public:
Permute(Graph* graph, const std::vector<uint32_t>& perm);
Transpose(Graph* graph, const std::vector<uint32_t>& perm);
protected:
std::vector<uint32_t> perm_;
@ -41,4 +41,4 @@ class Permute : public Operation {
} // namespace vx
} // namespace tim
#endif /* TIM_VX_OPS_PERMUTE_H_ */
#endif /* TIM_VX_OPS_TRANSPOSE_H_ */

4
src/tim/vx/ops/permute.cc → src/tim/vx/ops/transpose.cc
View File

@ -21,7 +21,7 @@
* DEALINGS IN THE SOFTWARE.
*
*****************************************************************************/
#include "tim/vx/ops/permute.h"
#include "tim/vx/ops/transpose.h"
#include "operation_private.h"
#include "vsi_nn_pub.h"
@ -30,7 +30,7 @@ namespace tim {
namespace vx {
namespace ops {
Permute::Permute(Graph* graph, const std::vector<uint32_t>& perm)
Transpose::Transpose(Graph* graph, const std::vector<uint32_t>& perm)
: Operation(graph, VSI_NN_OP_PERMUTE), perm_(std::move(perm)) {
this->impl()->node()->nn_param.permute.perm = perm_.data();
this->impl()->node()->nn_param.permute.dim_num = perm_.size();