Skip to content

Commit f17b1ca

Browse files
authored
[New Sample] Add Some Tabular Model Computational Graph (#141)
1 parent 31aaa7c commit f17b1ca

Some content is hidden

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

42 files changed

+19667
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
02456f748bbfa8c39ad3f211091f075c4953e0a3de4611edfdb244a1bbfd06bd
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": true
6+
}

samples/transformers-auto-model/sadhaklal_mlp-california-housing/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/sadhaklal_mlp-california-housing/input_tensor_constraints.py

Whitespace-only changes.
Lines changed: 65 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,65 @@
1+
import torch
2+
3+
4+
class GraphModule(torch.nn.Module):
5+
def forward(
6+
self,
7+
L_self_modules_fc1_parameters_weight_: torch.nn.parameter.Parameter,
8+
L_self_modules_fc1_parameters_bias_: torch.nn.parameter.Parameter,
9+
s77: torch.SymInt,
10+
L_x_: torch.Tensor,
11+
L_self_modules_fc2_parameters_weight_: torch.nn.parameter.Parameter,
12+
L_self_modules_fc2_parameters_bias_: torch.nn.parameter.Parameter,
13+
L_self_modules_fc3_parameters_weight_: torch.nn.parameter.Parameter,
14+
L_self_modules_fc3_parameters_bias_: torch.nn.parameter.Parameter,
15+
L_self_modules_fc4_parameters_weight_: torch.nn.parameter.Parameter,
16+
L_self_modules_fc4_parameters_bias_: torch.nn.parameter.Parameter,
17+
):
18+
l_self_modules_fc1_parameters_weight_ = L_self_modules_fc1_parameters_weight_
19+
l_self_modules_fc1_parameters_bias_ = L_self_modules_fc1_parameters_bias_
20+
l_x_ = L_x_
21+
l_self_modules_fc2_parameters_weight_ = L_self_modules_fc2_parameters_weight_
22+
l_self_modules_fc2_parameters_bias_ = L_self_modules_fc2_parameters_bias_
23+
l_self_modules_fc3_parameters_weight_ = L_self_modules_fc3_parameters_weight_
24+
l_self_modules_fc3_parameters_bias_ = L_self_modules_fc3_parameters_bias_
25+
l_self_modules_fc4_parameters_weight_ = L_self_modules_fc4_parameters_weight_
26+
l_self_modules_fc4_parameters_bias_ = L_self_modules_fc4_parameters_bias_
27+
linear = torch._C._nn.linear(
28+
l_x_,
29+
l_self_modules_fc1_parameters_weight_,
30+
l_self_modules_fc1_parameters_bias_,
31+
)
32+
l_x_ = (
33+
l_self_modules_fc1_parameters_weight_
34+
) = l_self_modules_fc1_parameters_bias_ = None
35+
act = torch.relu(linear)
36+
linear = None
37+
linear_1 = torch._C._nn.linear(
38+
act,
39+
l_self_modules_fc2_parameters_weight_,
40+
l_self_modules_fc2_parameters_bias_,
41+
)
42+
act = (
43+
l_self_modules_fc2_parameters_weight_
44+
) = l_self_modules_fc2_parameters_bias_ = None
45+
act_1 = torch.relu(linear_1)
46+
linear_1 = None
47+
linear_2 = torch._C._nn.linear(
48+
act_1,
49+
l_self_modules_fc3_parameters_weight_,
50+
l_self_modules_fc3_parameters_bias_,
51+
)
52+
act_1 = (
53+
l_self_modules_fc3_parameters_weight_
54+
) = l_self_modules_fc3_parameters_bias_ = None
55+
act_2 = torch.relu(linear_2)
56+
linear_2 = None
57+
linear_3 = torch._C._nn.linear(
58+
act_2,
59+
l_self_modules_fc4_parameters_weight_,
60+
l_self_modules_fc4_parameters_bias_,
61+
)
62+
act_2 = (
63+
l_self_modules_fc4_parameters_weight_
64+
) = l_self_modules_fc4_parameters_bias_ = None
65+
return (linear_3,)
Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,98 @@
1+
class Program_weight_tensor_meta_L_self_modules_fc1_parameters_weight_:
2+
name = "L_self_modules_fc1_parameters_weight_"
3+
shape = [50, 8]
4+
dtype = "torch.float32"
5+
device = "cpu"
6+
mean = -0.030
7+
std = 0.233
8+
data = None
9+
10+
11+
class Program_weight_tensor_meta_L_self_modules_fc1_parameters_bias_:
12+
name = "L_self_modules_fc1_parameters_bias_"
13+
shape = [50]
14+
dtype = "torch.float32"
15+
device = "cpu"
16+
mean = 0.027
17+
std = 0.228
18+
data = None
19+
20+
21+
class Program_weight_tensor_meta_s77:
22+
name = "s77"
23+
shape = []
24+
dtype = "torch.int64"
25+
device = "cpu"
26+
mean = None
27+
std = None
28+
data = [4]
29+
30+
31+
class Program_weight_tensor_meta_L_x_:
32+
name = "L_x_"
33+
shape = [3, 8]
34+
dtype = "torch.float32"
35+
device = "cpu"
36+
mean = -0.082
37+
std = 0.654
38+
data = None
39+
40+
41+
class Program_weight_tensor_meta_L_self_modules_fc2_parameters_weight_:
42+
name = "L_self_modules_fc2_parameters_weight_"
43+
shape = [50, 50]
44+
dtype = "torch.float32"
45+
device = "cpu"
46+
mean = 0.000
47+
std = 0.105
48+
data = None
49+
50+
51+
class Program_weight_tensor_meta_L_self_modules_fc2_parameters_bias_:
52+
name = "L_self_modules_fc2_parameters_bias_"
53+
shape = [50]
54+
dtype = "torch.float32"
55+
device = "cpu"
56+
mean = 0.010
57+
std = 0.115
58+
data = None
59+
60+
61+
class Program_weight_tensor_meta_L_self_modules_fc3_parameters_weight_:
62+
name = "L_self_modules_fc3_parameters_weight_"
63+
shape = [50, 50]
64+
dtype = "torch.float32"
65+
device = "cpu"
66+
mean = 0.011
67+
std = 0.121
68+
data = None
69+
70+
71+
class Program_weight_tensor_meta_L_self_modules_fc3_parameters_bias_:
72+
name = "L_self_modules_fc3_parameters_bias_"
73+
shape = [50]
74+
dtype = "torch.float32"
75+
device = "cpu"
76+
mean = 0.039
77+
std = 0.099
78+
data = None
79+
80+
81+
class Program_weight_tensor_meta_L_self_modules_fc4_parameters_weight_:
82+
name = "L_self_modules_fc4_parameters_weight_"
83+
shape = [1, 50]
84+
dtype = "torch.float32"
85+
device = "cpu"
86+
mean = 0.065
87+
std = 0.139
88+
data = None
89+
90+
91+
class Program_weight_tensor_meta_L_self_modules_fc4_parameters_bias_:
92+
name = "L_self_modules_fc4_parameters_bias_"
93+
shape = [1]
94+
dtype = "torch.float32"
95+
device = "cpu"
96+
mean = 0.070
97+
std = 0.000
98+
data = None
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
17586f580ad833be5fce667982761fc09e68d4e7e42ec7dd9a852ecb5a0c58ab
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": true
6+
}

samples/transformers-auto-model/sadhaklal_mlp-iris/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/sadhaklal_mlp-iris/input_tensor_constraints.py

Whitespace-only changes.

0 commit comments

Comments
 (0)