Home
last modified time | relevance | path

Searched defs:position_ids (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/dynamo/
H A Dtest_dynamo_with_onnxruntime_backend.py426 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 Drope.py138 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 Dtest_loop_ordering.py323 def forward(self, input_ids, labels, position_ids): argument