parent
d9ab299759
commit
0adce9a4fa
|
@ -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);
|
||||||
|
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue