Skip to content

Commit 9ea7cbc

Browse files
committed
[Add] paddle models
1 parent e4d1ac1 commit 9ea7cbc

File tree

350 files changed

+392464
-1
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

350 files changed

+392464
-1
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
class Program_weight_tensor_data_0:
2+
name = "input0"
3+
shape = [16, 2, 96]
4+
dtype = "float32"
5+
data = None
Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
import paddle
2+
3+
4+
class GraphModule(paddle.nn.Layer):
5+
def __init__(self):
6+
super().__init__()
7+
8+
def forward(
9+
self,
10+
input0: paddle.Tensor,
11+
t0: paddle.Tensor,
12+
t1: paddle.Tensor,
13+
t2: paddle.Tensor,
14+
):
15+
t3 = None
16+
# pd_op.matmul: (16x2x32xf32) <- (16x2x96xf32, 96x32xf32)
17+
t4 = paddle._C_ops.matmul(input0, t0, False, False)
18+
del input0, t0
19+
20+
# pd_op.add: (16x2x32xf32) <- (16x2x32xf32, 32xf32)
21+
t5 = paddle._C_ops.add(t4, t1)
22+
del t1
23+
24+
# pd_op.relu: (16x2x32xf32) <- (16x2x32xf32)
25+
t6 = paddle._C_ops.relu(t5)
26+
del t5
27+
28+
# pd_op.full: (1xf32) <- ()
29+
t7 = paddle._C_ops.full(
30+
[1], float("0.2"), paddle.float32, paddle.core.CPUPlace()
31+
)
32+
33+
# pd_op.dropout: (16x2x32xf32, 16x2x32xui8) <- (16x2x32xf32, None, 1xf32)
34+
t8, t9 = (lambda x, f: f(x))(
35+
paddle._C_ops.dropout(t6, None, t7, False, "upscale_in_train", 0, False),
36+
lambda out: out if isinstance(out, (list, tuple)) else (out, None),
37+
)
38+
39+
# pd_op.matmul: (16x2x16xf32) <- (16x2x32xf32, 32x16xf32)
40+
t10 = paddle._C_ops.matmul(t8, t2, False, False)
41+
del t2
42+
43+
return t4, t6, t7, t8, t9, t10
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
class Program_weight_tensor_parameter_3:
2+
name = "t0"
3+
shape = [96, 32]
4+
dtype = "float32"
5+
data = None
6+
7+
8+
class Program_weight_tensor_parameter_2:
9+
name = "t1"
10+
shape = [32]
11+
dtype = "float32"
12+
data = None
13+
14+
15+
class Program_weight_tensor_parameter_1:
16+
name = "t2"
17+
shape = [32, 16]
18+
dtype = "float32"
19+
data = None
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
{
2+
"framework": "paddle",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1
5+
}
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
class Program_weight_tensor_data_0:
2+
name = "input0"
3+
shape = [32, 3, 224, 224]
4+
dtype = "float32"
5+
data = None
6+
7+
8+
class Program_weight_tensor_data_1:
9+
name = "input1"
10+
shape = [1, 1, 768]
11+
dtype = "float32"
12+
data = None
13+
14+
15+
class Program_weight_tensor_data_2:
16+
name = "input2"
17+
shape = [1, 197, 768]
18+
dtype = "float32"
19+
data = None

0 commit comments

Comments
 (0)