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

Rename Unmaxpool2d to MaxUnpool2d 

Signed-off-by: zhao.xia <zhao.xia@verisilicon.com>
Signed-off-by: Kainan Cha <kainan.zha@verisilicon.com>
This commit is contained in:
zhao.xia 2021-05-28 10:38:07 +08:00 committed by Kainan Cha
parent 9c60671031
commit 26948d6646
4 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
include/tim/vx/ops/unmaxpool2d.h → include/tim/vx/ops/maxunpool2d.h
View File

@ -21,8 +21,8 @@
* DEALINGS IN THE SOFTWARE.
*
*****************************************************************************/
#ifndef TIM_VX_OPS_UNMAXPOOL2D_H_
#define TIM_VX_OPS_UNMAXPOOL2D_H_
#ifndef TIM_VX_OPS_MAXUNPOOL2D_H_
#define TIM_VX_OPS_MAXUNPOOL2D_H_
#include <array>
@ -34,7 +34,7 @@ namespace vx {
namespace ops {
/**
* ## UnMaxpool2d
* ## MaxUnpool2d
*
* Performs an 2-D Max pooling operation upsample
*
@ -42,9 +42,9 @@ namespace ops {
* - ksize : filter size.
*/
class UnMaxpool2d : public Operation {
class MaxUnpool2d : public Operation {
public:
UnMaxpool2d(Graph* graph, const std::array<uint32_t, 2>& ksize,
MaxUnpool2d(Graph* graph, const std::array<uint32_t, 2>& ksize,
const std::array<uint32_t, 2>& stride, DataLayout layout = DataLayout::WHCN);
protected:
@ -56,4 +56,4 @@ class UnMaxpool2d : public Operation {
} // namespace vx
} // namespace tim
#endif /* TIM_VX_OPS_UNMAXPOOL2D_H_ */
#endif /* TIM_VX_OPS_MAXUNPOOL2D_H_ */

2
src/tim/vx/ops/README.md
View File

@ -18,7 +18,7 @@ DeConv2d|DECONVOLUTION|Mapped|[tf.nn.conv2d_transpose](https://tensorflow.google
Reshape|RESHAPE|Mapped|[tf.reshape](https://tensorflow.google.cn/api_docs/python/tf/reshape)
Transpose|PERMUTE|Mapped|[tf.transpose](https://tensorflow.google.cn/api_docs/python/tf/transpose)
Prelu|PRELU|Mapped|[tf.keras.layers.PReLU](https://tensorflow.google.cn/api_docs/python/tf/keras/layers/PReLU)
UnMaxpool2d|UPSAMPLE|Mapped| Recover pixel from the outputs of MaxpoolWithArgmax.
MaxUnpool2d|UPSAMPLE|Mapped| Recover pixel from the outputs of MaxpoolWithArgmax.
Relu|RELU|Mapped|[tf.nn.relu](https://tensorflow.google.cn/api_docs/python/tf/nn/relu)
||RELUN|Deprecated|[tf.keras.layers.ReLU(max_value=N)](https://tensorflow.google.cn/api_docs/python/tf/keras/layers/ReLU)
Reorg|REORG|Mapped|[darknet.reorg](https://github.com/pjreddie/darknet/blob/master/src/reorg_layer.c)

4
src/tim/vx/ops/unmaxpool2d.cc → src/tim/vx/ops/maxunpool2d.cc
View File

@ -21,7 +21,7 @@
* DEALINGS IN THE SOFTWARE.
*
*****************************************************************************/
#include "tim/vx/ops/unmaxpool2d.h"
#include "tim/vx/ops/maxunpool2d.h"
#include "operation_private.h"
#include "type_utils.h"
@ -31,7 +31,7 @@ namespace tim {
namespace vx {
namespace ops {
UnMaxpool2d::UnMaxpool2d(Graph* graph, const std::array<uint32_t, 2>& ksize,
MaxUnpool2d::MaxUnpool2d(Graph* graph, const std::array<uint32_t, 2>& ksize,
const std::array<uint32_t, 2>& stride, DataLayout layout)
: Operation(graph, VSI_NN_OP_UPSAMPLE, 2, 1, layout),
ksize_(ksize), stride_(stride) {

10
src/tim/vx/ops/unmaxpool2d_test.cc → src/tim/vx/ops/maxunpool2d_test.cc
View File

@ -23,11 +23,11 @@
*****************************************************************************/
#include "tim/vx/context.h"
#include "tim/vx/graph.h"
#include "tim/vx/ops/unmaxpool2d.h"
#include "tim/vx/ops/maxunpool2d.h"
#include "gtest/gtest.h"
TEST(UnMaxpool2d, shape_2_2_1_fp32_kernel_2_stride_2) {
TEST(MaxUnpool2d, shape_2_2_1_fp32_kernel_2_stride_2) {
auto ctx = tim::vx::Context::Create();
auto graph = ctx->CreateGraph();
@ -59,7 +59,7 @@ TEST(UnMaxpool2d, shape_2_2_1_fp32_kernel_2_stride_2) {
EXPECT_TRUE(indices_tensor->CopyDataToTensor(indices.data(), indices.size()*4));
std::array<uint32_t, 2> ksize = {2, 2};
std::array<uint32_t, 2> stride = {2, 2};
auto op = graph->CreateOperation<tim::vx::ops::UnMaxpool2d>(ksize, stride);
auto op = graph->CreateOperation<tim::vx::ops::MaxUnpool2d>(ksize, stride);
(*op).BindInputs({values_tensor, indices_tensor}).BindOutputs({output_tensor});
EXPECT_TRUE(graph->Compile());
@ -70,7 +70,7 @@ TEST(UnMaxpool2d, shape_2_2_1_fp32_kernel_2_stride_2) {
EXPECT_EQ(golden, output);
}
TEST(UnMaxpool2d, shape_2_2_1_uint8_kernel_2_stride_2) {
TEST(MaxUnpool2d, shape_2_2_1_uint8_kernel_2_stride_2) {
auto ctx = tim::vx::Context::Create();
auto graph = ctx->CreateGraph();
@ -104,7 +104,7 @@ TEST(UnMaxpool2d, shape_2_2_1_uint8_kernel_2_stride_2) {
EXPECT_TRUE(indices_tensor->CopyDataToTensor(indices.data(), indices.size()));
std::array<uint32_t, 2> ksize = {2, 2};
std::array<uint32_t, 2> stride = {2, 2};
auto op = graph->CreateOperation<tim::vx::ops::UnMaxpool2d>(ksize, stride);
auto op = graph->CreateOperation<tim::vx::ops::MaxUnpool2d>(ksize, stride);
(*op).BindInputs({values_tensor, indices_tensor}).BindOutputs({output_tensor});
EXPECT_TRUE(graph->Compile());