Home
last modified time | relevance | path

Searched refs:read_tensor_from_checkpoint (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/demo/
H A Dtest_utils_test.py32 test_utils.read_tensor_from_checkpoint(checkpoint, 'int', tf.int32), 3)
34 test_utils.read_tensor_from_checkpoint(checkpoint, 'str', tf.string),
37 test_utils.read_tensor_from_checkpoint(checkpoint, 'list',
43 test_utils.read_tensor_from_checkpoint(checkpoint, 'str', tf.string)
48 test_utils.read_tensor_from_checkpoint(checkpoint, 'int', tf.string)
H A Dtest_utils.py35 def read_tensor_from_checkpoint(checkpoint: bytes, tensor_name: str, function
H A Dfederated_context_test.py240 test_utils.read_tensor_from_checkpoint(
270 test_utils.read_tensor_from_checkpoint(
H A Dserver_test.py234 result = test_utils.read_tensor_from_checkpoint(
H A Dplan_utils_test.py319 result = test_utils.read_tensor_from_checkpoint(checkpoint,
H A Daggregations_test.py146 test_utils.read_tensor_from_checkpoint(aggregate,