minor fix for msvc
Signed-off-by: raver119 <raver119@gmail.com>
This commit is contained in:
parent
d9ab299759
commit
0adce9a4fa
@ -656,7 +656,7 @@ TEST_F(DeclarableOpsTests9, concat_test19) {
|
||||
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.setIArguments(&axis, 1);
|
||||
|
||||
|
@ -36,5 +36,6 @@
|
||||
#include <GraphExecutioner.h>
|
||||
#include <gtest/gtest.h>
|
||||
#include <NDArrayFactory.h>
|
||||
#include <array>
|
||||
|
||||
#endif //LIBND4J_TESTLAYERS_H
|
||||
|
Loading…
x
Reference in New Issue
Block a user