Searched defs:embedding_rref (Results 1 – 1 of 1) sorted by relevance
2050 def _call_remote_embedding(cls, embedding_rref, input, offsets, per_sample_weights): argument2055 def _get_grad(cls, embedding_rref, context_id): argument