Searched refs:feature_out_dim (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/gnn/ |
| D | aggregator.py | 141 feature_out_dim, argument
150 self.out_dim = feature_out_dim
197 feature_out_dim, argument
206 feature_out_dim,
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_bgcf.py | 33 feature_out_dim, argument
38 … initializer("XavierUniform", [feature_in_dim * 2, feature_out_dim], dtype=mstype.float32))
62 feature_out_dim, argument
66 … initializer("XavierUniform", [feature_in_dim * 2, feature_out_dim], dtype=mstype.float32))
|