| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | make_batched_features_dataset_test.py | 41 def testRead(self, batch_size, num_epochs): argument
111 num_epochs): argument
139 self, batch_size, num_epochs): argument
172 def testParallelReadersAndParsers(self, batch_size, num_epochs, argument
208 def testDropFinalBatch(self, batch_size, num_epochs): argument
|
| H A D | make_csv_dataset_test.py | 36 def _make_csv_dataset(self, filenames, batch_size, num_epochs=1, **kwargs): argument
65 num_epochs): argument
81 num_epochs, argument
113 num_epochs=1, argument
|
| H A D | make_tf_record_dataset_test.py | 42 def testRead(self, batch_size, num_epochs, file_index, num_parallel_reads, argument
82 def testShuffle(self, batch_size, num_epochs, num_parallel_reads, seed): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | tf_record_test_base.py | 41 num_epochs, argument
162 num_epochs, argument
219 num_epochs=1, argument
269 def _next_expected_batch(self, file_indices, batch_size, num_epochs, argument
301 def _verify_records(self, outputs, batch_size, file_index, num_epochs, argument
|
| H A D | fixed_length_record_dataset_test.py | 81 def dataset_fn(filenames, num_epochs, batch_size=None): argument
233 def _build_dataset(self, num_epochs, compression_type=None): argument
|
| H A D | tf_record_dataset_test.py | 39 num_epochs=1, argument
196 num_epochs, argument
|
| H A D | checkpoint_test.py | 209 def _build_graph(start, stop, num_epochs): argument
257 def _build_graph(start, stop, num_epochs): argument
|
| H A D | text_line_dataset_test.py | 90 def dataset_fn(filenames, num_epochs, batch_size=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | readers.py | 218 dataset, num_epochs, shuffle, shuffle_buffer_size, shuffle_seed): argument
230 num_epochs=None, argument
337 num_epochs=None, # TODO(aaudibert): Change default to 1 when graduating. argument
641 num_epochs=None, argument
922 num_epochs=None, argument
1106 num_epochs=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | input.py | 80 def limit_epochs(tensor, num_epochs=None, name=None): argument
120 num_epochs=None, argument
207 num_epochs=None, argument
280 def range_input_producer(limit, num_epochs=None, shuffle=True, seed=None, argument
323 def slice_input_producer(tensor_list, num_epochs=None, shuffle=True, seed=None, argument
|
| /aosp_15_r20/external/executorch/extension/training/examples/XOR/ |
| H A D | train.cpp | 85 size_t num_epochs = 5000; in main() local
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | training_loss.py | 56 backend, train_dataloader, eval_dataloader, model, optimizer, num_epochs, evaluation argument
|
| /aosp_15_r20/external/libopus/dnn/torch/neural-pitch/ |
| H A D | training.py | 99 num_epochs = args.epochs variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| H A D | quantization_mnist_test.py | 190 def _Run(self, is_training, use_trt, batch_size, num_epochs, model_dir): argument
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_model_report_fx.py | 524 def run_model_and_common_checks(self, model, ex_input, num_epochs, batch_size): argument
|
| /aosp_15_r20/external/federated-compute/fcp/protos/ |
| H A D | plan.proto | 796 string num_epochs = 3; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_partition.cc | 890 const int num_epochs = 100; in AddControlEdges() local
|
| /aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/ |
| D | plan.pb.h | 15448 inline const std::string& DatasetInputPlaceholders::num_epochs() const { in num_epochs() function
15479 inline void DatasetInputPlaceholders::set_allocated_num_epochs(std::string* num_epochs) { in set_allocated_num_epochs()
|