Searched defs:position_ids (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/test/onnx/dynamo/ |
H A D | test_dynamo_with_onnxruntime_backend.py | 426 def forward(self, hidden_states, attention_mask, position_ids): argument 522 def forward(self, hidden_states, attention_mask, position_ids): argument 608 def forward(self, input_ids, attention_mask, position_ids): argument
|
/aosp_15_r20/external/executorch/examples/models/llama/ |
H A D | rope.py | 138 def hf_apply_rotary_emb(q, k, cos, sin, position_ids=None, unsqueeze_dim=1): argument
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_loop_ordering.py | 323 def forward(self, input_ids, labels, position_ids): argument
|