Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py508 def _build_multidim_tensor(dim, dim_size, value=None, dtype=torch.float): function
3652 _build_multidim_tensor(
3659 _build_multidim_tensor(
3668 _build_multidim_tensor(