From de41aa80cb91c30d8ac4a7617b9f937be47f1481 Mon Sep 17 00:00:00 2001
From: Li Xiaoquan <xiaoquan.li@denglin.ai>
Date: Mon, 24 Dec 2018 15:52:22 +0800
Subject: [PATCH] [NNVM] Fix dtype of output of pad.

  Dtype of output of pad should follows input, but if dtype of input is not float,
  output will still be float becase pad_value is float.
---
 nnvm/src/top/nn/nn.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nnvm/src/top/nn/nn.cc b/nnvm/src/top/nn/nn.cc
index 09dfbb211f00..e301f167ff1d 100644
--- a/nnvm/src/top/nn/nn.cc
+++ b/nnvm/src/top/nn/nn.cc
@@ -620,7 +620,8 @@ NNVM_REGISTER_OP(pad)
     for (size_t i = 0; i < pad_width.ndim(); ++i) {
       pad_after.push_back(tvm::make_const(tvm::Int(32), pad_width[i][1]));
     }
-    return Array<Tensor>{ topi::pad(inputs[0], pad_before, pad_after, param.pad_value) };
+    return Array<Tensor>{ topi::pad(inputs[0], pad_before, pad_after,
+                          tvm::make_const(inputs[0]->dtype, param.pad_value)) };
 })
 .set_support_level(1);