| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_outside_compilation_test.py | 139 def train_step(): function
159 def train_step(): function
180 def train_step(): function
206 def train_step(): function
233 def train_step(): function
259 def train_step(): function
286 def train_step(): function
312 def train_step(): function
423 def train_step(): function
446 def train_step(): function
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | custom_training_loop_input_test.py | 109 def train_step(): function
127 def train_step(data): function
147 def train_step(data): function
189 def train_step(data): function
207 def train_step(data): function
226 def train_step(data): function
281 def train_step(data): function
306 def train_step(_): function
938 def train_step(data): function
971 def train_step(input_iterator): function
[all …]
|
| H A D | tpu_strategy_test.py | 354 def train_step(): function
472 def train_step(): function
522 def train_step(): function
537 def train_step(): function
574 def train_step(): function
742 def train_step(data): function
800 def train_step(): function
816 def train_step(): function
836 def train_step(): function
851 def train_step(): function
|
| H A D | custom_training_loop_gradient_test.py | 73 def train_step(x): function
100 def train_step(x): function
134 def train_step(): function
|
| H A D | zero_batch_test.py | 126 def train_step(): function
183 def train_step(iterator): function
|
| H A D | mirrored_variable_test.py | 632 def train_step(): function
|
| H A D | parameter_server_strategy_test.py | 751 def train_step(data): function
|
| H A D | distribute_lib_test.py | 527 def train_step(input_data): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | remote_test.py | 256 def train_step(iterator): function
279 def train_step(iterator): function
305 def train_step(iterator): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | const_op_test.py | 95 def train_step(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/ |
| H A D | tpu_memory_test.py | 143 def train_step(iterator): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| H A D | mnist_train.py | 138 def train_step(features): function
|
| /aosp_15_r20/external/pytorch/functorch/examples/compilation/ |
| H A D | linear_train.py | 73 def train(train_step, weights): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| H A D | failure_handler_test.py | 163 def train_step(): function
|
| H A D | gce_failure_handler_test.py | 166 def train_step(): function
|
| /aosp_15_r20/external/pytorch/scripts/release_notes/ |
| H A D | classifier.py | 266 def train_step(batch, model, optimizer, loss): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sdca_ops.cc | 172 auto train_step = [&](const int64_t begin, const int64_t end) { in DoCompute() local
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/ |
| H A D | test_2d_composability.py | 257 def train_step( function
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_training.py | 1042 def train_step( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training.py | 765 def train_step(self, data): member in Model
|