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,
|