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;
|
Nd4jLong *newShape;
|
||||||
COPY_SHAPE(inShape, newShape);
|
COPY_SHAPE(inShape, newShape);
|
||||||
|
|
||||||
return SHAPELIST(newShape);
|
return SHAPELIST(CONSTANT(newShape));
|
||||||
}
|
}
|
||||||
|
|
||||||
DECLARE_TYPES(strided_slice) {
|
DECLARE_TYPES(strided_slice) {
|
||||||
|
|
Loading…
Reference in New Issue