Home
last modified time | relevance | path

Searched refs:sparse_rpc_format_to_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dutils.py19 def sparse_rpc_format_to_tensor(sparse_rpc_format): function
56 tensor = sparse_rpc_format_to_tensor(tensor)
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/server/
H A Dserver.py7 from utils import sparse_rpc_format_to_tensor, sparse_tensor_to_rpc_format
263 gradient = sparse_rpc_format_to_tensor(gradient)