diff --git a/include/tim/vx/ops/permute.h b/include/tim/vx/ops/transpose.h similarity index 88% rename from include/tim/vx/ops/permute.h rename to include/tim/vx/ops/transpose.h index fe33e68..ee6f18f 100644 --- a/include/tim/vx/ops/permute.h +++ b/include/tim/vx/ops/transpose.h @@ -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& perm); + Transpose(Graph* graph, const std::vector& perm); protected: std::vector perm_; @@ -41,4 +41,4 @@ class Permute : public Operation { } // namespace vx } // namespace tim -#endif /* TIM_VX_OPS_PERMUTE_H_ */ \ No newline at end of file +#endif /* TIM_VX_OPS_TRANSPOSE_H_ */ \ No newline at end of file diff --git a/src/tim/vx/ops/permute.cc b/src/tim/vx/ops/transpose.cc similarity index 93% rename from src/tim/vx/ops/permute.cc rename to src/tim/vx/ops/transpose.cc index 228e92b..eb35180 100644 --- a/src/tim/vx/ops/permute.cc +++ b/src/tim/vx/ops/transpose.cc @@ -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& perm) +Transpose::Transpose(Graph* graph, const std::vector& 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();