Searched defs:forward_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | transforms.py | 200 def forward_shape(self, shape): member in Transform
270 def forward_shape(self, shape): member in _InverseTransform
382 def forward_shape(self, shape): member in ComposeTransform
466 def forward_shape(self, shape): member in IndependentTransform
519 def forward_shape(self, shape): member in ReshapeTransform
597 def forward_shape(self, shape): member in PowerTransform
800 def forward_shape(self, shape): member in AffineTransform
872 def forward_shape(self, shape): member in CorrCholeskyTransform
917 def forward_shape(self, shape): member in SoftmaxTransform
973 def forward_shape(self, shape): member in StickBreakingTransform
|