Searched defs:embedding_cols (Results 1 – 1 of 1) sorted by relevance
65 int64_t embedding_cols = qweight.size(1); in prepack() local244 const int32_t embedding_cols = weight_sizes[cols_dim]; in qembeddingbag_byte_prepack_out() local336 const int32_t embedding_cols = weight_sizes[cols_dim]; in qembeddingbag_byte_prepack_meta() local367 int64_t embedding_cols = weight.size(1); in _qembeddingbag_nbit_prepack_helper() local