Searched refs:write_metrics_to_checkpoint (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/federated-compute/fcp/artifact_building/ |
| H A D | checkpoint_utils_test.py | 54 write_metrics_to_checkpoint=True,
64 state_vars, metrics_vars, write_metrics_to_checkpoint argument
66 del state_vars, metrics_vars, write_metrics_to_checkpoint
77 write_metrics_to_checkpoint=True,
93 write_metrics_to_checkpoint=True,
|
| H A D | checkpoint_utils.py | 37 write_metrics_to_checkpoint: bool = True,
104 state_vars, metric_vars, write_metrics_to_checkpoint
108 if has_metrics and write_metrics_to_checkpoint:
120 state_vars, None, write_metrics_to_checkpoint
|
| H A D | federated_compute_plan_builder.py | 727 write_metrics_to_checkpoint: bool = True,
810 write_metrics_to_checkpoint=write_metrics_to_checkpoint,
1107 if write_metrics_to_checkpoint:
1486 write_metrics_to_checkpoint: bool = True,
1604 write_metrics_to_checkpoint,
|