minor fix for msvc

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2019-08-21 16:34:07 +03:00
parent d9ab299759
commit 0adce9a4fa
2 changed files with 2 additions and 1 deletions

View File

@ -656,7 +656,7 @@ TEST_F(DeclarableOpsTests9, concat_test19) {
context.setInputArray(e, array, true); context.setInputArray(e, array, true);
} }
auto z = NDArrayFactory::create<float>('c', {arrays.size(), 5, 20}); auto z = NDArrayFactory::create<float>('c', {(Nd4jLong) arrays.size(), 5, 20});
context.setOutputArray(0, &z, false); context.setOutputArray(0, &z, false);
context.setIArguments(&axis, 1); context.setIArguments(&axis, 1);

View File

@ -36,5 +36,6 @@
#include <GraphExecutioner.h> #include <GraphExecutioner.h>
#include <gtest/gtest.h> #include <gtest/gtest.h>
#include <NDArrayFactory.h> #include <NDArrayFactory.h>
#include <array>
#endif //LIBND4J_TESTLAYERS_H #endif //LIBND4J_TESTLAYERS_H