strided_slice_bp shape fn leak fix

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2019-07-24 17:11:29 +03:00 committed by AlexDBlack
parent ac1fc1a27a
commit ce0743da17
1 changed files with 1 additions and 1 deletions

View File

@ -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) {