strided_slice_bp shape fn leak fix
Signed-off-by: raver119 <raver119@gmail.com>master
parent
ac1fc1a27a
commit
ce0743da17
|
@ -640,7 +640,7 @@ namespace nd4j {
|
|||
Nd4jLong *newShape;
|
||||
COPY_SHAPE(inShape, newShape);
|
||||
|
||||
return SHAPELIST(newShape);
|
||||
return SHAPELIST(CONSTANT(newShape));
|
||||
}
|
||||
|
||||
DECLARE_TYPES(strided_slice) {
|
||||
|
|
Loading…
Reference in New Issue