Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_type_promotion.py126 def make_scalar_tensor(dtype): function
152 s = make_scalar_tensor(dtype=torch.chalf)
176 s = make_scalar_tensor(dtype=torch.cfloat)
195 s = make_scalar_tensor(dtype=torch.cdouble)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py6368make_scalar_tensor = partial(make_tensor, (), device='cpu', dtype=dtype, requires_grad=requires_gr…
6394 ((3, 4), make_scalar_tensor(), make_scalar_tensor()),