Searched defs:train_step_fn (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | basic_loops.py | 22 train_step_fn, argument
|
| /aosp_15_r20/external/pytorch/functorch/examples/ensembling/ |
| H A D | parallel_train.py | 82 def train_step_fn(weights, batch, targets, lr=0.2): function
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 4399 def train_step_fn(use_transform, weights, batch, targets, lr=0.2): function
4503 def train_step_fn(weights, batch, targets, lr): function
|