| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_outside_compilation_test.py | 141 def train_step(): function
161 def train_step(): function
182 def train_step(): function
208 def train_step(): function
235 def train_step(): function
261 def train_step(): function
288 def train_step(): function
314 def train_step(): function
425 def train_step(): function
448 def train_step(): function
[all …]
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | custom_training_loop_test.py | 76 def train_step(x): function
100 def train_step(x, y): function
123 def train_step(x, y): function
150 def test_eager_and_tf_function(self, train_step): argument
161 def train_step(x): function
187 def train_step(x): function
224 def train_step(x): function
|
| D | add_loss_correctness_test.py | 47 def train_step(x, y, w=None): function
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_models_test.py | 76 def train_step(iterator): function
101 def train_step(replicated_inputs): function
139 def train_step(iterator): function
164 def train_step(iterator): function
196 def train_step(iterator): function
239 def train_step(input_iterator): function
348 def train_step(iterator): function
375 def train_step(iterator): function
405 def train_step(iterator): function
480 def train_step(iterator): function
|
| D | custom_training_loop_metrics_test.py | 49 def train_step(): function
|
| D | mirrored_strategy_test.py | 109 def train_step(iterator): function
|
| D | ctl_correctness_test.py | 176 def train_step(dist_inputs): function
|
| D | distribute_strategy_test.py | 2391 def train_step(dist_inputs): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_gradient_test.py | 77 def train_step(x): function
104 def train_step(x): function
138 def train_step(): function
|
| D | custom_training_loop_input_test.py | 111 def train_step(): function
129 def train_step(data): function
149 def train_step(data): function
191 def train_step(data): function
209 def train_step(data): function
228 def train_step(data): function
283 def train_step(data): function
308 def train_step(_): function
910 def train_step(data): function
943 def train_step(input_iterator): function
[all …]
|
| D | zero_batch_test.py | 129 def train_step(): function
186 def train_step(iterator): function
|
| D | tpu_strategy_test.py | 217 def train_step(): function
335 def train_step(): function
385 def train_step(): function
400 def train_step(): function
437 def train_step(): function
605 def train_step(data): function
663 def train_step(): function
679 def train_step(): function
|
| D | parameter_server_strategy_test.py | 755 def train_step(data): function
|
| D | distribute_lib_test.py | 510 def train_step(input_data): function
|
| /external/tensorflow/tensorflow/python/keras/premade/ |
| D | wide_deep.py | 112 def train_step(self, data): member in WideDeepModel
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_test.py | 216 def train_step(iterator): function
239 def train_step(iterator): function
265 def train_step(iterator): function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | mnist_train.py | 142 def train_step(features): function
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | tpu_strategy_test.py | 165 def train_step(iterator): function
|
| D | multi_worker_tutorial_test.py | 262 def train_step(iterator): function
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 67 def train_step(self, inputs, model, loss_fn, optimizer, batch_size): member in CustomMnistBenchmark
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils_test.py | 234 def train_step(data, labels): function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 172 auto train_step = [&](const int64 begin, const int64 end) { in DoCompute() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 766 def train_step(self, data): member in Model
|
| D | training_test.py | 1612 def train_step(self, data): member in TrainingTest.test_single_input_no_tuple_wrapping.MyModel
|