diff --git a/include/tim/vx/ops/batch2space.h b/include/tim/vx/ops/batch2space.h
index cd9257c..0ff34b3 100644
--- a/include/tim/vx/ops/batch2space.h
+++ b/include/tim/vx/ops/batch2space.h
@@ -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);
 
diff --git a/include/tim/vx/ops/space2batch.h b/include/tim/vx/ops/space2batch.h
index 484c9b4..9237c4a 100644
--- a/include/tim/vx/ops/space2batch.h
+++ b/include/tim/vx/ops/space2batch.h
@@ -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);
 
diff --git a/src/tim/vx/ops/batch2space.cc b/src/tim/vx/ops/batch2space.cc
index 2eddc2e..df74766 100644
--- a/src/tim/vx/ops/batch2space.cc
+++ b/src/tim/vx/ops/batch2space.cc
@@ -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),
diff --git a/src/tim/vx/ops/space2batch.cc b/src/tim/vx/ops/space2batch.cc
index a171a2f..60ed966 100644
--- a/src/tim/vx/ops/space2batch.cc
+++ b/src/tim/vx/ops/space2batch.cc
@@ -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),