| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | input_lib_type_spec_test.py | 52 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
84 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
117 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
163 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
203 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
227 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
263 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
336 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
411 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
465 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
[all …]
|
| H A D | input_lib_test.py | 288 strategy_combinations.mirrored_strategy_with_gpu_and_cpu
366 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
421 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
555 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
707 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
799 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
827 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
859 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
896 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
936 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
[all …]
|
| H A D | strategy_combinations.py | 388 mirrored_strategy_with_gpu_and_cpu = combinations.NamedDistribution( variable
521 mirrored_strategy_with_gpu_and_cpu,
529 mirrored_strategy_with_gpu_and_cpu,
545 mirrored_strategy_with_gpu_and_cpu,
562 mirrored_strategy_with_gpu_and_cpu,
591 mirrored_strategy_with_gpu_and_cpu, mirrored_strategy_with_two_gpus
|
| H A D | values_test.py | 59 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
233 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
256 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
289 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
478 strategy_combinations.mirrored_strategy_with_gpu_and_cpu
511 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
527 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | mirrored_strategy_test.py | 68 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
358 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
393 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
527 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
693 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
945 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
989 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
1064 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
1393 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | checkpoint_utils_test.py | 68 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
104 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | strategy_combinations_test.py | 103 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
157 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | vars_test.py | 71 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
419 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
472 distribution=[strategy_combinations.mirrored_strategy_with_gpu_and_cpu],
1126 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
1194 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | checkpointing_test.py | 33 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | warm_starting_util_test.py | 46 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | distributed_variable_test.py | 64 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
76 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | mirrored_values_test.py | 71 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | moving_averages_test.py | 33 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | distribute_utils_test.py | 150 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | metrics_v1_test.py | 76 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | strategy_gather_test.py | 53 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | mirrored_variable_test.py | 64 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | strategy_common_test.py | 263 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| H A D | custom_training_loop_input_test.py | 244 strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | optimizer_combinations.py | 82 strategy_combinations_base.mirrored_strategy_with_gpu_and_cpu,
95 strategy_combinations_base.mirrored_strategy_with_gpu_and_cpu,
108 strategy_combinations_base.mirrored_strategy_with_gpu_and_cpu,
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.__internal__.distribute.combinations.pbtxt | 24 name: "mirrored_strategy_with_gpu_and_cpu"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | cross_device_ops_test.py | 404 distribution=strategy_combinations.mirrored_strategy_with_gpu_and_cpu,
|