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

Change back the inferface name (#44) 

Signed-off-by: yuenan.li <yuenan.li@verisilicon.com>

Co-authored-by: yuenan.li <yuenan.li@verisilicon.com>
This commit is contained in:
liyuenan 2021-05-14 20:08:53 +08:00 committed by GitHub
parent 56bd7bf8c8
commit 55ef50385e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/tim/vx/ops/batch2space.h
View File

@ -32,9 +32,9 @@ namespace tim {
namespace vx {
namespace ops {
class BatchToSpace : public Operation {
class Batch2Space : public Operation {
public:
BatchToSpace(Graph* graph, const std::vector<int>& block_size,
Batch2Space(Graph* graph, const std::vector<int>& block_size,
const std::vector<int>& crop,
DataLayout layout = DataLayout::WHCN);

4
include/tim/vx/ops/space2batch.h
View File

@ -32,9 +32,9 @@ namespace tim {
namespace vx {
namespace ops {
class SpaceToBatch : public Operation {
class Space2Batch : public Operation {
public:
SpaceToBatch(Graph* graph, const std::vector<int>& block_size,
Space2Batch(Graph* graph, const std::vector<int>& block_size,
const std::vector<int>& pad,
DataLayout layout = DataLayout::WHCN);

2
src/tim/vx/ops/batch2space.cc
View File

@ -30,7 +30,7 @@ namespace tim {
namespace vx {
namespace ops {
BatchToSpace::BatchToSpace(Graph* graph, const std::vector<int>& block_size,
Batch2Space::Batch2Space(Graph* graph, const std::vector<int>& block_size,
const std::vector<int>& crop, DataLayout layout)
: Operation(graph, VSI_NN_OP_BATCH2SPACE, 0, 0, layout),
block_size_(block_size),

2
src/tim/vx/ops/space2batch.cc
View File

@ -30,7 +30,7 @@ namespace tim {
namespace vx {
namespace ops {
SpaceToBatch::SpaceToBatch(Graph* graph, const std::vector<int>& block_size,
Space2Batch::Space2Batch(Graph* graph, const std::vector<int>& block_size,
const std::vector<int>& pad, DataLayout layout)
: Operation(graph, VSI_NN_OP_SPACE2BATCH, 0, 0, layout),
block_size_(block_size),