reshape validation fix
Signed-off-by: raver119 <raver119@gmail.com>
This commit is contained in:
parent
b32dd1bf92
commit
70e08c3a6c
@ -164,6 +164,7 @@ namespace nd4j {
|
||||
|
||||
// we can launch op using Int arguments
|
||||
if (inputShape->size() == 1) {
|
||||
REQUIRE_TRUE(block.numI() > 0, 0, "Reshape: new shape should be provided as NDArray or int arguments, but nothing was defined");
|
||||
std::vector<int> *arguments = block.getIArguments();
|
||||
|
||||
int e = 1;
|
||||
|
Loading…
x
Reference in New Issue
Block a user