Searched refs:all_gather_into_tensor_coalesced (Results 1 – 9 of 9) sorted by relevance
244 rs = c10d.all_gather_into_tensor_coalesced(285 rs = _c10d.all_gather_into_tensor_coalesced(
52 return torch.ops._c10d_functional.all_gather_into_tensor_coalesced(
360 def all_gather_into_tensor_coalesced( function381 …tensor_list = torch.ops._c10d_functional.all_gather_into_tensor_coalesced( # type: ignore[attr-de…
12 all_gather_into_tensor_coalesced,266 outputs = torch.ops._c10d_functional.all_gather_into_tensor_coalesced(285 outputs = all_gather_into_tensor_coalesced(677 ag0 = funcol.all_gather_into_tensor_coalesced(args, "0")
314 res = ft_c.all_gather_into_tensor_coalesced(tensors, mesh)510 res = ft_c.all_gather_into_tensor_coalesced(tensors, mesh)
990 ar = torch.ops.c10d_functional.all_gather_into_tensor_coalesced(1042 tensor_list = torch.ops.c10d_functional.all_gather_into_tensor_coalesced(
168 std::vector<at::Tensor> all_gather_into_tensor_coalesced( in all_gather_into_tensor_coalesced() function192 return all_gather_into_tensor_coalesced( in all_gather_into_tensor()352 ::all_gather_into_tensor_coalesced), in TORCH_LIBRARY()
33 funcol_native.all_gather_into_tensor_coalesced: funcol_py.all_gather_into_tensor_coalesced,
6363 @register_lowering(_c10d_functional.all_gather_into_tensor_coalesced)6368 _c10d_functional.all_gather_into_tensor_coalesced.default,