From 4bb40baae09e4baf637ae2673d156a28546a2d0c Mon Sep 17 00:00:00 2001
From: Howave <myhouseng@gmail.com>
Date: Fri, 26 Oct 2018 00:29:00 +0800
Subject: [PATCH] [DOCS] Fix C++ example:graph_runtime.cc:151: Check failed:
data->ndim == data_out->ndim (2 vs. 1) (#1987)
---
docs/deploy/nnvm.md | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/docs/deploy/nnvm.md b/docs/deploy/nnvm.md
index e4ce14528..1e0d17f8b 100644
--- a/docs/deploy/nnvm.md
+++ b/docs/deploy/nnvm.md
@@ -96,8 +96,8 @@ int main()
run();
DLTensor* y;
- int out_ndim = 1;
- int64_t out_shape[1] = {1000, };
+ int out_ndim = 2;
+ int64_t out_shape[2] = {1, 1000, };
TVMArrayAlloc(out_shape, out_ndim, dtype_code, dtype_bits, dtype_lanes, device_type, device_id, &y);
// get the function from the module(get output data)
--
GitLab