Searched defs:embedding_matrix (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/libtextclassifier/native/lang_id/common/ |
| H A D | embedding-network.cc | 159 const EmbeddingNetworkParams::Matrix &embedding_matrix = in ConcatEmbeddings() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset18.py | 231 embedding_matrix, argument
|
| H A D | symbolic_opset10.py | 592 embedding_matrix, argument
|
| H A D | symbolic_opset11.py | 1205 embedding_matrix, argument
|
| H A D | symbolic_helper.py | 1942 embedding_matrix, argument
|
| H A D | symbolic_opset9.py | 933 embedding_matrix, argument
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 7657 def forward(self, embedding_matrix, input, offset, weights): argument
7686 def forward(self, embedding_matrix, input, weights): argument
7714 def forward(self, embedding_matrix, input, weights, offsets): argument
7755 def forward(self, embedding_matrix, input, weights): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 10094 def embedding_norm(input, embedding_matrix, max_norm): argument
10098 def embedding_norm_script(input, embedding_matrix, max_norm): argument
|