Searched defs:embedding_dims (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_pruning_op.py | 24 def _test_rowwise_prune_op(self, embedding_rows, embedding_dims, indices_type, weights_dtype): argument
67 def test_rowwise_prune_op_32bit_indices(self, embedding_rows, embedding_dims, weights_dtype): argument
79 def test_rowwise_prune_op_64bit_indices(self, embedding_rows, embedding_dims, weights_dtype): argument
|
| /aosp_15_r20/external/libtextclassifier/native/lang_id/common/ |
| H A D | embedding-feature-extractor.cc | 41 const std::string embedding_dims = in Setup() local
|