(self, inputs, index, size=None)
| 31 | self.aggr = aggr |
| 32 | |
| 33 | def __call__(self, inputs, index, size=None): |
| 34 | size = tf.reduce_max(index) + 1 if size is None else size |
| 35 | |
| 36 | gate = self.gate_nn(inputs) |
| 37 | inputs = self.nn(inputs) if self.nn is not None else inputs |
| 38 | |
| 39 | gate = mp_ops.scatter_softmax(gate, index, size=size) |
| 40 | outputs = mp_ops.scatter_(self.aggr, gate * inputs, index, size=size) |
| 41 | return outputs |
nothing calls this directly
no outgoing calls
no test coverage detected