Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py3543 include_2d_tensor=False, argument
3610 if include_2d_tensor:
4359 include_2d_tensor=True, # (B, *)
4487 include_2d_tensor=True, # (B, *)
4525 include_2d_tensor=True, # (B, *)
4579 include_2d_tensor=True, # (B, *)