Searched defs:LongTensor (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/runner/ |
| H A D | eager.py | 41 tokens: Optional[torch.LongTensor] = None,
42 input_pos: Optional[torch.LongTensor] = None,
43 mask: Optional[torch.LongTensor] = None,
|
| H A D | exported.py | 43 tokens: Optional[torch.LongTensor] = None,
44 input_pos: Optional[torch.LongTensor] = None,
45 mask: Optional[torch.LongTensor] = None,
|
| H A D | native.py | 53 mask: Optional[torch.LongTensor] = None,
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | llama_transformer.py | 488 tokens: Optional[torch.LongTensor] = None, # tokens
490 torch.LongTensor
|
| /aosp_15_r20/external/executorch/examples/models/phi-3-mini/ |
| H A D | phi_3_mini.py | 33 input_ids: torch.LongTensor = None,
|
| /aosp_15_r20/external/executorch/extension/evalue_util/test/ |
| H A D | print_evalue_test.cpp | 228 TEST(PrintEvalueTest, LongTensor) { in TEST() argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 3714 def check_subclass_warn(input: torch.LongTensor) -> torch.LongTensor:
|