| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | cluster_coordinator_test.py | 540 def input_fn(): function
568 def input_fn(): function
613 def input_fn(): function
638 def input_fn(): function
652 def input_fn(): function
742 def input_fn(): function
803 def input_fn(): function
1007 def input_fn(): function
1056 def input_fn(): function
1386 def input_fn(): function
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | sparse_test.py | 312 def testSerializeDeserialize(self, input_fn): argument
328 def testSerializeManyDeserialize(self, input_fn): argument
|
| H A D | random_seed_test.py | 118 def testRandomSeed(self, input_fn, output_fn): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_api.py | 188 def input_fn(mod, inputs, device_mesh): function
233 def input_fn(inputs, device_mesh): function
|
| H A D | test_optimizers.py | 38 def input_fn(mod, inputs, device_mesh): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | distributed_training_utils_v1.py | 837 def distributed_function(input_fn): argument
852 def execution_function(input_fn): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | async_checkpoint_test.py | 52 def input_fn(params): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v2_optimizer_test.py | 84 def input_fn(ctx): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | ram_file_system_test.py | 126 def input_fn(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | distribute_lib.py | 973 def make_input_fn_iterator(self, argument
1903 def make_input_fn_iterator(self, # pylint: disable=useless-super-delegation argument
2284 def _make_input_fn_iterator(self, input_fn, replication_mode): argument
3666 def _make_input_fn_iterator(self, argument
|
| H A D | parameter_server_strategy_test.py | 518 def _test_input_fn_iterator(self, argument
|
| H A D | collective_all_reduce_strategy.py | 704 def _make_input_fn_iterator( argument
|
| H A D | one_device_strategy.py | 294 def _make_input_fn_iterator( argument
|
| H A D | parameter_server_strategy.py | 354 def _make_input_fn_iterator( argument
|
| H A D | collective_all_reduce_strategy_test.py | 238 def _test_input_fn_iterator(self, argument
|
| H A D | distribute_lib_test.py | 80 def _make_input_fn_iterator( argument
|
| H A D | mirrored_strategy.py | 549 def _make_input_fn_iterator( argument
|
| H A D | strategy_test_lib.py | 357 def _test_input_fn_iterable( argument
|
| H A D | tpu_strategy.py | 944 def _make_input_fn_iterator( argument
|
| H A D | input_lib_test.py | 274 def _create_dataset_or_input_fn(self, input_type, input_fn): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | accumulate_n_benchmark.py | 115 def _RunBenchmark(self, tag, input_fn, sizes, ninputs, repeats): argument
|
| /aosp_15_r20/external/cronet/testing/rust_gtest_interop/ |
| H A D | gtest_attribute.rs | 120 let mut input_fn = parse_macro_input!(input as ItemFn); in gtest() localVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | input_lib.py | 277 def __init__(self, input_fn, input_workers, input_contexts, strategy): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| H A D | trt_convert.py | 1411 def build(self, input_fn): argument
|