Searched refs:feature_in_dim (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/gnn/ |
| D | aggregator.py | 140 feature_in_dim, argument
149 self.in_dim = feature_in_dim
196 feature_in_dim, argument
205 feature_in_dim,
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_bgcf.py | 32 feature_in_dim, argument
38 … initializer("XavierUniform", [feature_in_dim * 2, feature_out_dim], dtype=mstype.float32))
61 feature_in_dim, argument
66 … initializer("XavierUniform", [feature_in_dim * 2, feature_out_dim], dtype=mstype.float32))
|