Searched defs:get_embedding (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_pytorch_onnx_onnxruntime.py | 11168 def get_embedding(embedding_dim: int): member in TestONNXRuntime.test_set_attr_modules.InnerModule2 |