Searched defs:init_rpc (Results 1 – 3 of 3) sorted by relevance
38 def init_rpc(self): member in ShardedTensorTestBase53 def init_comms(self, init_rpc=True, backend="nccl"): argument83 def with_comms(func=None, init_rpc=True, backend="nccl"): argument
84 def init_rpc( function
50 def init_rpc(name, backend): function