Searched defs:col_dim (Results 1 – 4 of 4) sorted by relevance
122 const auto col_dim = packed_weight_sizes.size() - 1; in qembeddingbag_byte_unpack_out() local172 const auto col_dim = packed_weight_sizes.size() - 1; in qembeddingbag_byte_unpack_meta() local
85 col_dim, argument
227 int64_t col_dim = dim_nums.lhs_batching_dimensions().size() + 1; in GemmOpConversionRewrite() local
174 const auto col_dim = packed_weight_sizes.size() - 1; in qembeddingbag_byte_unpack() local