From ce0743da1795044fb82ed33415d65e0044dd1c9a Mon Sep 17 00:00:00 2001 From: raver119 Date: Wed, 24 Jul 2019 17:11:29 +0300 Subject: [PATCH] strided_slice_bp shape fn leak fix Signed-off-by: raver119 --- .../include/ops/declarable/generic/parity_ops/strided_slice.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libnd4j/include/ops/declarable/generic/parity_ops/strided_slice.cpp b/libnd4j/include/ops/declarable/generic/parity_ops/strided_slice.cpp index 22837b92a..44fe5999a 100644 --- a/libnd4j/include/ops/declarable/generic/parity_ops/strided_slice.cpp +++ b/libnd4j/include/ops/declarable/generic/parity_ops/strided_slice.cpp @@ -640,7 +640,7 @@ namespace nd4j { Nd4jLong *newShape; COPY_SHAPE(inShape, newShape); - return SHAPELIST(newShape); + return SHAPELIST(CONSTANT(newShape)); } DECLARE_TYPES(strided_slice) {