Skip to content

Commit 26b99b6

Browse files
committed
Add cv samples for models with more than 30 subgraphs.
1 parent 7f3cf4c commit 26b99b6

File tree

1,480 files changed

+200309
-0
lines changed

Some content is hidden

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

1,480 files changed

+200309
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
1c86568233f9b834312d9525c485088ea33394123460e0ad7ddb2644d1bc2a7e
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "paddle",
3+
"model_name": "Cascade-FasterRCNN-ResNet50-FPN",
4+
"num_devices_required": 1,
5+
"num_nodes_required": 1
6+
}
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
class Program_weight_tensor_data_0:
2+
name = "data_0"
3+
shape = [196192]
4+
dtype = "int32"
5+
min_val = 0
6+
max_val = 196191
7+
data = None
8+
9+
10+
class Program_weight_tensor_data_1:
11+
name = "data_1"
12+
shape = []
13+
dtype = "int64"
14+
data = [253]
15+
16+
17+
class Program_weight_tensor_data_2:
18+
name = "data_2"
19+
shape = [196192]
20+
dtype = "int32"
21+
min_val = 0
22+
max_val = 196415
23+
data = None
24+
25+
26+
class Program_weight_tensor_data_3:
27+
name = "data_3"
28+
shape = []
29+
dtype = "int64"
30+
data = [3]
Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
import paddle
2+
3+
4+
class GraphModule(paddle.nn.Layer):
5+
def __init__(self):
6+
super().__init__()
7+
8+
def forward(self, data_0, data_1, data_2, data_3):
9+
# pd_op.full_int_array: (1xi64) <- ()
10+
full_int_array_0 = [0]
11+
12+
# builtin.combine: ([xi64]) <- (xi64)
13+
combine_0 = [data_1]
14+
del data_1
15+
16+
# pd_op.stack: (1xi64) <- ([xi64])
17+
stack_0 = paddle._C_ops.stack(combine_0, 0)
18+
del combine_0
19+
20+
# pd_op.slice: (-1xi32) <- (196192xi32, 1xi64, 1xi64)
21+
slice_0 = paddle._C_ops.slice(data_0, [0], full_int_array_0, stack_0, [-1], [])
22+
del data_0, full_int_array_0, stack_0
23+
24+
# pd_op.full: (1xi32) <- ()
25+
full_0 = paddle._C_ops.full(
26+
[1], float("0"), paddle.int32, paddle.core.CPUPlace()
27+
)
28+
29+
# pd_op.gather: (-1xi32) <- (196192xi32, -1xi32, 1xi32)
30+
gather_0 = paddle._C_ops.gather(data_2, slice_0, full_0)
31+
del data_2, full_0, slice_0
32+
33+
# pd_op.full: (xi64) <- ()
34+
full_1 = paddle._C_ops.full(
35+
[], float("0"), paddle.int64, paddle.framework._current_expected_place()
36+
)
37+
38+
# pd_op.equal: (xb) <- (xi64, xi64)
39+
equal_0 = paddle._C_ops.equal(data_3, full_1)
40+
del data_3, full_1
41+
42+
return equal_0, gather_0
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
00ef08cc89ec8f8051c57321c9dd142e65268c6a6cab8a869a81bcd6f9d10a0d
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "paddle",
3+
"model_name": "Cascade-FasterRCNN-ResNet50-FPN",
4+
"num_devices_required": 1,
5+
"num_nodes_required": 1
6+
}
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 = "data_0"
3+
shape = [2]
4+
dtype = "int32"
5+
data = [0, 1]
6+
7+
8+
class Program_weight_tensor_data_1:
9+
name = "data_1"
10+
shape = []
11+
dtype = "int64"
12+
data = [2]
13+
14+
15+
class Program_weight_tensor_data_2:
16+
name = "data_2"
17+
shape = [2]
18+
dtype = "int32"
19+
data = [189304, 189448]
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
import paddle
2+
3+
4+
class GraphModule(paddle.nn.Layer):
5+
def __init__(self):
6+
super().__init__()
7+
8+
def forward(self, data_0, data_1, data_2):
9+
# pd_op.full_int_array: (1xi64) <- ()
10+
full_int_array_0 = [0]
11+
12+
# builtin.combine: ([xi64]) <- (xi64)
13+
combine_0 = [data_1]
14+
del data_1
15+
16+
# pd_op.stack: (1xi64) <- ([xi64])
17+
stack_0 = paddle._C_ops.stack(combine_0, 0)
18+
del combine_0
19+
20+
# pd_op.slice: (-1xi32) <- (-1xi32, 1xi64, 1xi64)
21+
slice_0 = paddle._C_ops.slice(data_0, [0], full_int_array_0, stack_0, [-1], [])
22+
del data_0, full_int_array_0, stack_0
23+
24+
# pd_op.full: (1xi32) <- ()
25+
full_0 = paddle._C_ops.full(
26+
[1], float("0"), paddle.int32, paddle.core.CPUPlace()
27+
)
28+
29+
# pd_op.gather: (-1xi32) <- (-1xi32, -1xi32, 1xi32)
30+
gather_0 = paddle._C_ops.gather(data_2, slice_0, full_0)
31+
del data_2, full_0, slice_0
32+
33+
return gather_0
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+

0 commit comments

Comments
 (0)