Searched defs:input_columns (Results 1 – 2 of 2) sorted by relevance
28 const auto input_columns = packed_weight.size(1); in unpack() local124 const int32_t input_columns = packed_weight_sizes[col_dim]; in qembeddingbag_byte_unpack_out() local173 const auto input_columns = packed_weight_sizes[col_dim]; in qembeddingbag_byte_unpack_meta() local189 const auto input_columns = packed_weight.size(1); in _qembeddingbag_nbit_unpack_helper() local
176 const int32_t input_columns = packed_weight_sizes[col_dim]; in qembeddingbag_byte_unpack() local547 const auto input_columns = packed_weight.size(1); in qembeddingbag_4bit_unpack() local