Home
last modified time | relevance | path

Searched defs:agent_rref (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/rl/
H A Dagent.py98 def select_action_batch(agent_rref, observer_id, state): argument
139 def select_action_non_batch(agent_rref, observer_id, state): argument
H A Dobserver.py50 def run_ob_episode(self, agent_rref, n_steps): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/
H A Dreinforcement_learning_rpc_test.py105 def run_episode(self, agent_rref, n_steps): argument