Home
last modified time | relevance | path

Searched refs:save_tff_structure_to_checkpoint (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/artifact_building/
H A Dcheckpoint_utils_test.py252 checkpoint_utils.save_tff_structure_to_checkpoint(
273 checkpoint_utils.save_tff_structure_to_checkpoint(
297 checkpoint_utils.save_tff_structure_to_checkpoint(
H A Dcheckpoint_utils.py478 def save_tff_structure_to_checkpoint( function
/aosp_15_r20/external/federated-compute/fcp/demo/
H A Dfederated_context.py290 checkpoint_utils.save_tff_structure_to_checkpoint(state, var_names, tmpfile)