Searched refs:checkpoint_type (Results 1 – 3 of 3) sorted by relevance
14 name = "checkpoint_type",15 srcs = ["checkpoint_type.py"],83 ":checkpoint_type",
30 from fcp.artifact_building import checkpoint_type731 …experimental_client_update_format: checkpoint_type.CheckpointFormatType = checkpoint_type.Checkpoi…900 == checkpoint_type.CheckpointFormatType.APPEND_SLICES_MERGE_READ1179 …experimental_checkpoint_write: checkpoint_type.CheckpointFormatType = checkpoint_type.CheckpointFo…1331 checkpoint_type.CheckpointFormatType.APPEND_SLICES_MERGE_WRITE,1332 checkpoint_type.CheckpointFormatType.APPEND_SLICES_MERGE_READ,1350 == checkpoint_type.CheckpointFormatType.APPEND_SLICES_MERGE_WRITE1362 == checkpoint_type.CheckpointFormatType.TF1_SAVE_SLICES1484 …experimental_client_checkpoint_write: checkpoint_type.CheckpointFormatType = checkpoint_type.Check…
371 def test_fake_tensor_mode_simple(self, checkpoint_type): argument400 if checkpoint_type == "state_dict":415 elif checkpoint_type == "checkpoint_file":