Searched defs:is_embedding_op (Results 1 – 2 of 2) sorted by relevance
233 bool is_embedding_op) { in embedding_bag_nbit_impl()405 bool is_embedding_op) { in embedding_bag_byte_impl()566 bool is_embedding_op) { in embedding_bag_byte_helper()664 bool is_embedding_op) { in _embedding_bag_nbit_helper()768 bool is_embedding_op) { in embeddingbag_byte()789 bool is_embedding_op) { in embeddingbag_4bit()
198 bool is_embedding_op) { in embedding_bag_byte_impl()277 bool is_embedding_op = false; in embedding_bag_byte_rowwise_offsets() local