diff --git a/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/__init__.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/__init__.py new file mode 100644 index 0000000000..5682c52104 --- /dev/null +++ b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/__init__.py @@ -0,0 +1,15 @@ +import os +import glob + +# 获取当前文件所在目录 +current_dir = os.path.dirname(__file__) + +# 获取当前目录下所有的 .py 文件路径 +py_files = glob.glob(os.path.join(current_dir, "*.py")) + +# 动态导入所有 .py 文件 +for py_file in py_files: + # 获取文件名(不含扩展名) + module_name = os.path.basename(py_file)[:-3] + # 导入模块 + __import__('layerApicase.inplace_stragtegy.' + module_name, globals(), locals(), []) diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/erfinv__base_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/erfinv__base_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/erfinv__base_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/erfinv__base_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/exponential__base_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/exponential__base_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/exponential__base_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/exponential__base_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/relu__func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/relu__func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/relu__func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/relu__func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__0_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__0_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__0_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__0_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__1_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__1_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__1_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__1_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__2_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__2_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__2_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__2_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__3_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__3_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__3_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__3_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__base_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__base_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/reshape__base_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/reshape__base_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__0_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__0_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__0_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__0_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__1_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__1_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__1_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__1_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__2_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__2_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__2_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__2_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__base_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__base_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/squeeze__base_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/squeeze__base_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__0_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__0_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__0_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__0_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__1_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__1_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__1_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__1_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__2_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__2_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__2_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__2_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__base_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__base_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/tanh__base_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/tanh__base_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/unsqueeze__0_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/unsqueeze__0_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/unsqueeze__0_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/unsqueeze__0_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/unsqueeze__base_func.py b/framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/unsqueeze__base_func.py similarity index 100% rename from framework/e2e/PaddleLT_new/layerApicase/math_sublayer/unsqueeze__base_func.py rename to framework/e2e/PaddleLT_new/layerApicase/inplace_stragtegy/unsqueeze__base_func.py diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_0_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_0_func.py index 39c49bc1d9..0545d7a283 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_0_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_0_func.py @@ -16,6 +16,7 @@ def forward(self, ): forward """ out = paddle.broadcast_shape( x_shape=[2, 1, 3, 1, 4, 2, 3], y_shape=[1, 2, 1, 1, 3], ) + out = paddle.to_tensor(out, stop_gradient=False) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_base_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_base_func.py index 11bd9ee26a..b8c66fd2fa 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_base_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/broadcast_shape_base_func.py @@ -16,6 +16,7 @@ def forward(self, ): forward """ out = paddle.broadcast_shape( x_shape=[2, 3, 4, 4, 1], y_shape=[2, 3, 1, 4, 3], ) + out = paddle.to_tensor(out, stop_gradient=False) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_0_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_0_func.py index 767a497c3e..a951ebf63c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_0_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_0_func.py @@ -15,7 +15,7 @@ def forward(self, x, y, ): """ forward """ - out = paddle.dist(x, y, p=-inf, ) + out = paddle.dist(x, y, p=float("-inf"), ) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_1_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_1_func.py index 48c06f734f..f8783a5d6c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_1_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/dist_1_func.py @@ -15,7 +15,7 @@ def forward(self, x, y, ): """ forward """ - out = paddle.dist(x, y, p=inf, ) + out = paddle.dist(x, y, p=float("inf"), ) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_1_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_1_func.py index ad48c80dc3..3e5ec4690b 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_1_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_1_func.py @@ -15,7 +15,7 @@ def forward(self, x, ): """ forward """ - out = paddle.median(x, axis=1, keepdims=True, ) + out = paddle.median(x, axis=1, keepdim=True, ) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_base_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_base_func.py index 50af28262c..3eac794360 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_base_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/median_base_func.py @@ -15,7 +15,7 @@ def forward(self, x, ): """ forward """ - out = paddle.median(x, axis=0, keepdims=False, ) + out = paddle.median(x, axis=0, keepdim=False, ) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_1_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_1_func.py index a8cdf148f9..255b3259fb 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_1_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_1_func.py @@ -15,7 +15,7 @@ def forward(self, arr, ): """ forward """ - out = paddle.put_along_axis(arr, indices=paddle.to_tensor([[[[2]]]], dtype='int32', stop_gradient=False), values=-4.0, axis=2, reduce='multiple', ) + out = paddle.put_along_axis(arr, indices=paddle.to_tensor([[[[2]]]], dtype='int32', stop_gradient=False), values=-4.0, axis=2, reduce='multiply', ) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_base_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_base_func.py index 44e117b9ef..3a521be5c0 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_base_func.py +++ b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/put_along_axis_base_func.py @@ -15,7 +15,7 @@ def forward(self, arr, ): """ forward """ - out = paddle.put_along_axis(arr, indices=paddle.to_tensor([[[[0]]]], dtype='int32', stop_gradient=False), value=21.0, axis=0, ) + out = paddle.put_along_axis(arr, indices=paddle.to_tensor([[[[0]]]], dtype='int32', stop_gradient=False), values=21.0, axis=0, ) return out diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_0_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_0_class.py index e98e5ce0a9..d0afb69662 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_0_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_0_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_2_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_2_class.py index 9c8242f309..02bfa69bee 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_2_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_2_class.py @@ -20,18 +20,23 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ - paddle tensor + numpy array """ - inputs = () + inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) return inputs - def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]]).astype('float32'), ) return inputs - diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_3_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_3_class.py index dc66cb8df8..6a23e2a1ea 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_3_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_3_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False),) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[0.6964692], [0.28613934]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_4_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_4_class.py index b1d606b37e..8c9fec195c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_4_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_4_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_5_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_5_class.py index ad330511c6..b8e58f9666 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_5_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_5_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[0.6964692], [0.28613934]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_6_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_6_class.py index 6fc116b343..5fad87e013 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_6_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_6_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_7_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_7_class.py index 0b16f14fea..45672f92f7 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_7_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_7_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[0.6964692], [0.28613934]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_8_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_8_class.py index 9e15e57df7..194ca4ab8c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_8_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_8_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[[0.6964692, 0.28613934, 0.22685145]], [[0.5513148, 0.71946895, 0.42310646]]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_9_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_9_class.py index 8fdd48c25f..1e1773a6ac 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_9_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_9_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_base_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_base_class.py index 7c392c3ef4..4b54c6a32a 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_base_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_base_class.py @@ -20,11 +20,18 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor """ - inputs = () + inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) return inputs @@ -32,6 +39,6 @@ def create_numpy_inputs(): """ numpy array """ - inputs = (paddle.to_tensor([[0.6964692], [0.28613934]], dtype='float32', stop_gradient=False), ) + inputs = (np.array([[0.6964692], [0.28613934]]).astype('float32'), ) return inputs diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_1_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_1_class.py index 2685a7dcfe..3249e8f1a6 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_1_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_1_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_2_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_2_class.py index d5a98a2fac..d3f3cd4b9c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_2_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_2_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_3_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_3_class.py index 22e341d37a..ab1f40ca60 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_3_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_3_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_4_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_4_class.py index 52b78a99c4..23f836053c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_4_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_4_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_7_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_7_class.py index e22c302e4d..3aaa69ef09 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_7_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_7_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_8_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_8_class.py index 94711232d9..6740cd8bd8 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_8_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_8_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_base_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_base_class.py index 14e452c6fa..e952b14dd0 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_base_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_base_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_1_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_1_class.py index da9938d2a9..b0bd796bfa 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_1_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_1_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_2_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_2_class.py index 18695f4b69..394473b3e8 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_2_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_2_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_3_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_3_class.py index 21f5615268..b0b4458a6c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_3_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_3_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_4_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_4_class.py index b7a74a380f..4da6b57612 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_4_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_4_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_5_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_5_class.py index 61cebed0e7..c12085cb54 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_5_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_5_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_base_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_base_class.py index db33c19990..4249ea7c1f 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_base_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_base_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_0_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_0_class.py index 753e73b774..4f02581ef4 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_0_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_0_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_10_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_10_class.py index 2dfe5a6464..63ce4909ff 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_10_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_10_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_11_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_11_class.py index 7819267df9..5457d2a430 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_11_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_11_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_12_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_12_class.py index c7f1fb0f53..c4c9d814b8 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_12_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_12_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_13_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_13_class.py index 8e55fbf354..e2a3f163bc 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_13_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_13_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_15_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_15_class.py index 1fdeb40a13..ebfa465e72 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_15_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_15_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_17_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_17_class.py index f362d473b0..131f4c59ec 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_17_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_17_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_18_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_18_class.py index 26d0da684f..38e27da78c 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_18_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_18_class.py @@ -20,6 +20,12 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_1_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_1_class.py index 1184c0e3c8..d395804189 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_1_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_1_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_2_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_2_class.py index 29670ce2bf..8f6bd450c4 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_2_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_2_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_3_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_3_class.py index 0701152666..fe47c599ea 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_3_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_3_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_4_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_4_class.py index 7caeb8dba7..96e0c9af69 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_4_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_4_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_7_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_7_class.py index 4373e7554e..8f3d31519d 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_7_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_7_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_8_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_8_class.py index 196c09918c..59004984a7 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_8_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_8_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1, -1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_base_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_base_class.py index 8a55479f8e..5bfb844034 100644 --- a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_base_class.py +++ b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm_base_class.py @@ -20,6 +20,13 @@ def forward(self, data, ): return out +def create_inputspec(): + inputspec = ( + paddle.static.InputSpec(shape=(-1, -1), dtype=paddle.float32, stop_gradient=False), + ) + return inputspec + + def create_tensor_inputs(): """ paddle tensor