Searched refs:sample_inputs_embedding (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | functorch_additional_op_db.py | 194 def sample_inputs_embedding(op_info, device, dtype, requires_grad, **kwargs): function
271 sample_inputs_func=sample_inputs_embedding,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 8124 def sample_inputs_embedding(op_info, device, dtype, requires_grad, **kwargs): function
20490 sample_inputs_func=sample_inputs_embedding,
|