From fb8d09a812332dc42dbb35cbb57c5b6f882ba022 Mon Sep 17 00:00:00 2001 From: Unknown Date: Fri, 23 Mar 2018 18:11:10 +0800 Subject: [PATCH] fix conflicts --- mace/core/operator.cc | 4 +- mace/ops/space_to_depth_test.cc | 77 --------------------------------- 2 files changed, 1 insertion(+), 80 deletions(-) delete mode 100644 mace/ops/space_to_depth_test.cc diff --git a/mace/core/operator.cc b/mace/core/operator.cc index 41524be9..1a88eacf 100644 --- a/mace/core/operator.cc +++ b/mace/core/operator.cc @@ -1,5 +1,4 @@ -// -// Copyright (c) 2017 XiaoMi All rights reserved. + // #include @@ -63,7 +62,6 @@ std::unique_ptr OperatorRegistry::CreateOperator( } namespace ops { - // Keep in lexicographical order extern void Register_Activation(OperatorRegistry *op_registry); extern void Register_AddN(OperatorRegistry *op_registry); diff --git a/mace/ops/space_to_depth_test.cc b/mace/ops/space_to_depth_test.cc deleted file mode 100644 index 37d020a9..00000000 --- a/mace/ops/space_to_depth_test.cc +++ /dev/null @@ -1,77 +0,0 @@ -// -// Copyright (c) 2017 XiaoMi All rights reserved. -// - -#include "mace/core/operator.h" -#include "mace/ops/ops_test_util.h" - -namespace mace { -namespace ops { -namespace test { - -class SpaceToDepthOpTest : public OpsTestBase {}; - -TEST_F(SpaceToDepthOpTest, C8G4_CPU) { - // Construct graph - OpsTestNet net; - OpDefBuilder("SpaceToDepth", "SpaceToDepthTest") - .Input("Input") - .Output("Output") - .AddIntArg("block_size", 2) - .Finalize(net.NewOperatorDef()); - - // Add input data - net.AddInputFromArray( - "Input", {1, 2, 4, 4}, - {0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23, - 8, 9, 10, 11, 12, 13, 14, 15, 24, 25, 26, 27, 28, 29, 30, 31}); - - // Run - net.RunOp(); - - // Check - auto expected = CreateTensor( - {1, 1, 2, 16}, - {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, - 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31}); - - ExpectTensorNear(*expected, *net.GetOutput("Output"), 0.001); -} - -TEST_F(SpaceToDepthOpTest, C16G4_OPENCL) { - // Construct graph - OpsTestNet net; - - // Add input data - net.AddInputFromArray( - "Input", {1, 2, 4, 4}, - {0, 1, 2, 3, 4, 5, 6, 7, 16, 17, 18, 19, 20, 21, 22, 23, - 8, 9, 10, 11, 12, 13, 14, 15, 24, 25, 26, 27, 28, 29, 30, 31}); - BufferToImage(&net, "Input", "InputImage", - kernels::BufferType::IN_OUT_CHANNEL); - - OpDefBuilder("SpaceToDepth", "SpaceToDepthTest") - .Input("InputImage") - .Output("OutputImage") - .AddIntArg("block_size", 2) - .Finalize(net.NewOperatorDef()); - - // Run - net.RunOp(DeviceType::OPENCL); - - // Transfer output - ImageToBuffer(&net, "OutputImage", "Output", - kernels::BufferType::IN_OUT_CHANNEL); - - // Check - auto expected = CreateTensor( - {1, 1, 2, 16}, - {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, - 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31}); - - ExpectTensorNear(*expected, *net.GetOutput("Output"), 0.001); -} - -} // namespace test -} // namespace ops -} // namespace mace -- GitLab