Searched refs:GNNFeatureTransform (Results 1 – 1 of 1) sorted by relevance
26 class GNNFeatureTransform(nn.Cell): class75 super(GNNFeatureTransform, self).__init__()156 self.fc = GNNFeatureTransform(self.in_dim,268 self.feature_transform = GNNFeatureTransform(273 self.f_1_transform = GNNFeatureTransform(276 self.f_2_transform = GNNFeatureTransform(289 self.residual_transform = GNNFeatureTransform(