Searched refs:canonical_weight_shapes (Results 1 – 2 of 2) sorted by relevance
275 def canonical_weight_shapes(self): member in _CudnnRNN347 for sp in self.canonical_weight_shapes
1048 sum(map(np.prod, rnn.canonical_weight_shapes)) +