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. * DEALINGS IN THE SOFTWARE.
* *
*****************************************************************************/ *****************************************************************************/
#ifndef TIM_VX_OPS_PERMUTE_H_ #ifndef TIM_VX_OPS_TRANSPOSE_H_
#define TIM_VX_OPS_PERMUTE_H_ #define TIM_VX_OPS_TRANSPOSE_H_
#include "tim/vx/operation.h" #include "tim/vx/operation.h"
namespace tim { namespace tim {
namespace vx { namespace vx {
namespace ops { namespace ops {
class Permute : public Operation { class Transpose : public Operation {
public: public:
Permute(Graph* graph, const std::vector<uint32_t>& perm); Transpose(Graph* graph, const std::vector<uint32_t>& perm);
protected: protected:
std::vector<uint32_t> perm_; std::vector<uint32_t> perm_;
@ -41,4 +41,4 @@ class Permute : public Operation {
} // namespace vx } // namespace vx
} // namespace tim } // 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. * DEALINGS IN THE SOFTWARE.
* *
*****************************************************************************/ *****************************************************************************/
#include "tim/vx/ops/permute.h" #include "tim/vx/ops/transpose.h"
#include "operation_private.h" #include "operation_private.h"
#include "vsi_nn_pub.h" #include "vsi_nn_pub.h"
@ -30,7 +30,7 @@ namespace tim {
namespace vx { namespace vx {
namespace ops { 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)) { : 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.perm = perm_.data();
this->impl()->node()->nn_param.permute.dim_num = perm_.size(); this->impl()->node()->nn_param.permute.dim_num = perm_.size();