Searched refs:include_2d_tensor (Results 1 – 1 of 1) sorted by relevance
3543 include_2d_tensor=False, argument3610 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, *)