Searched refs:run_computation (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/federated-compute/fcp/demo/ |
| H A D | federated_context_test.py | 210 async def test_invoke_success(self, run_computation): argument
211 run_computation.return_value = build_result_checkpoint(7)
224 run_computation.assert_called_once_with(
232 plan = run_computation.call_args.args[2]
241 run_computation.call_args.args[3], input_var_names[0], tf.int32), 3)
244 async def test_invoke_with_value_reference(self, run_computation): argument
245 run_computation.side_effect = [
268 self.assertEqual(run_computation.call_count, 2)
271 run_computation.call_args.args[3], input_var_names[0], tf.int32),
285 async def test_invoke_with_run_computation_error(self, run_computation): argument
[all …]
|
| H A D | server_test.py | 192 self.server.run_computation(
250 self.server.run_computation(
|
| H A D | server.py | 79 async def run_computation( member in InProcessServer
|
| H A D | federated_context.py | 241 return await self._server.run_computation(
|