Searched refs:read_tensor_from_checkpoint (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/federated-compute/fcp/demo/ |
H A D | test_utils_test.py | 32 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 D | test_utils.py | 35 def read_tensor_from_checkpoint(checkpoint: bytes, tensor_name: str, function
|
H A D | federated_context_test.py | 240 test_utils.read_tensor_from_checkpoint( 270 test_utils.read_tensor_from_checkpoint(
|
H A D | server_test.py | 234 result = test_utils.read_tensor_from_checkpoint(
|
H A D | plan_utils_test.py | 319 result = test_utils.read_tensor_from_checkpoint(checkpoint,
|
H A D | aggregations_test.py | 146 test_utils.read_tensor_from_checkpoint(aggregate,
|