Searched defs:forward_backward (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/extension/training/pybindings/ |
| H A D | _training_module.py | 33 def forward_backward(self, method_name: str, inputs: Sequence[Any]) -> List[Any]: member in TrainingModule
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | function.py | 573 def forward_backward(self, num_doutputs=None): member in _DelayedRewriteGradientFunctions
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 3965 def forward_backward(cuda, rnn, input_val, grad_output, weights_val, hx_val, grad_hy, function
|