| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_outside_compilation_test.py | 140 def train_step(): function
160 def train_step(): function
181 def train_step(): function
207 def train_step(): function
234 def train_step(): function
260 def train_step(): function
287 def train_step(): function
313 def train_step(): function
424 def train_step(): function
447 def train_step(): function
[all …]
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | custom_training_loop_test.py | 72 def train_step(x): function
96 def train_step(x, y): function
119 def train_step(x, y): function
146 def test_eager_and_tf_function(self, train_step): argument
157 def train_step(x): function
183 def train_step(x): function
220 def train_step(x): function
|
| D | add_loss_correctness_test.py | 43 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
166 def train_step(iterator): function
198 def train_step(iterator): function
241 def train_step(input_iterator): function
350 def train_step(iterator): function
377 def train_step(iterator): function
407 def train_step(iterator): function
482 def train_step(iterator): function
|
| D | custom_training_loop_metrics_test.py | 46 def train_step(): function
|
| D | mirrored_strategy_test.py | 105 def train_step(iterator): function
|
| D | multi_worker_test.py | 251 def train_step(iterator): function
|
| D | ctl_correctness_test.py | 184 def train_step(dist_inputs): function
|
| D | distribute_strategy_test.py | 2388 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 | 113 def train_step(): function
131 def train_step(data): function
151 def train_step(data): function
193 def train_step(data): function
211 def train_step(data): function
230 def train_step(data): function
285 def train_step(data): function
310 def train_step(_): function
942 def train_step(data): function
975 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 | 270 def train_step(): function
388 def train_step(): function
438 def train_step(): function
453 def train_step(): function
490 def train_step(): function
658 def train_step(data): function
716 def train_step(): function
732 def train_step(): function
752 def train_step(): function
767 def train_step(): function
|
| D | mirrored_variable_test.py | 636 def train_step(): function
|
| D | parameter_server_strategy_test.py | 755 def train_step(data): function
|
| D | distribute_lib_test.py | 530 def train_step(input_data): function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | const_op_test.py | 100 def train_step(): function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_test.py | 217 def train_step(iterator): function
240 def train_step(iterator): function
266 def train_step(iterator): function
|
| /external/tensorflow/tensorflow/python/keras/premade/ |
| D | wide_deep.py | 107 def train_step(self, data): member in WideDeepModel
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | mnist_train.py | 142 def train_step(features): 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 | 229 def train_step(data, labels): function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 172 auto train_step = [&](const int64_t begin, const int64_t end) { in DoCompute() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 767 def train_step(self, data): member in Model
|
| D | training_test.py | 1607 def train_step(self, data): member in TrainingTest.test_single_input_no_tuple_wrapping.MyModel
|