diff --git a/docs/deploy/nnvm.md b/docs/deploy/nnvm.md index 4040de35ea545..650912231b12e 100644 --- a/docs/deploy/nnvm.md +++ b/docs/deploy/nnvm.md @@ -59,9 +59,11 @@ An example in c++. #include #include +#include #include #include -#include +#include +#include int main() { @@ -97,7 +99,9 @@ int main() int64_t in_shape[4] = {1, 3, 224, 224}; TVMArrayAlloc(in_shape, in_ndim, dtype_code, dtype_bits, dtype_lanes, device_type, device_id, &x); // load image data saved in binary - std::ifstream data_fin("cat.bin", std::ios::binary); + const std::string data_filename = "cat.bin"; + std::ifstream data_fin(data_filename, std::ios::binary); + if(!data_fin) throw std::runtime_error("Could not open: " + data_filename); data_fin.read(static_cast(x->data), 3 * 224 * 224 * 4); // get the function from the module(set input data)