Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py9139 def _make_test_tensor(shape, contiguous=True): function
9168 _test_smooth_l1_loss_vs_huber_loss_helper(input=_make_test_tensor(shape),
9169 target=_make_test_tensor(shape),
9175 _test_smooth_l1_loss_vs_huber_loss_helper(input=_make_test_tensor(shape),
9176 target=_make_test_tensor(shape),
9181 … _test_smooth_l1_loss_vs_huber_loss_helper(input=_make_test_tensor(shape, contiguous=False),
9182 … target=_make_test_tensor(shape, contiguous=False),