| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | mirrored_strategy_test.py | 362 def model_fn(): function
400 def model_fn(): function
422 def model_fn(): function
436 def model_fn(): function
458 def model_fn(): function
475 def model_fn(): function
494 def model_fn(): function
507 def model_fn(): function
535 def model_fn(): function
553 def model_fn(): function
[all …]
|
| H A D | mirrored_variable_test.py | 105 def model_fn(): function
123 def model_fn(): function
139 def model_fn(): function
153 def model_fn(): function
164 def model_fn(): function
178 def model_fn(): function
199 def model_fn(): function
213 def model_fn(): function
252 def model_fn(): function
429 def model_fn(name): function
[all …]
|
| H A D | strategy_common_test.py | 102 def model_fn(): function
113 def model_fn(): function
124 def model_fn(): function
136 def model_fn(): function
165 def model_fn(): function
177 def model_fn(): function
190 def model_fn(): function
206 def model_fn(): function
|
| H A D | parameter_server_strategy_test.py | 131 def model_fn(): function
237 def model_fn(): function
276 def model_fn(): function
382 def model_fn(): function
|
| H A D | single_loss_example.py | 61 def model_fn(x): function
98 def model_fn(x): function
|
| H A D | collective_all_reduce_strategy_test.py | 218 def model_fn(): function
|
| H A D | strategy_test_lib.py | 438 def model_fn(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | gradients_test.py | 165 def model_fn(activation): function
189 def model_fn(inps, init_state): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | async_checkpoint_test.py | 61 def model_fn(features, labels, mode, params): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | remapper_test.py | 101 def _VerifyValues(self, model_fn, use_low_precision, fused_op, epilog_ops): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | ram_file_system_test.py | 112 def model_fn(features, labels, mode, params): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| H A D | quantize_model_test.py | 950 def model_fn(self, x: core.Tensor) -> Mapping[str, core.Tensor]: member in StaticRangeQuantizationTest.test_model_ptq_with_uncalibrated_subgraph.IfModel
|