Searched refs:segmentDims (Results 1 – 1 of 1) sorted by relevance
2251 std::vector<dim_t> segmentDims(offsets.dims().begin(), offsets.dims().end()); in createEmbeddingBagByteRowwiseOffsets() local2255 segmentDims[0] -= 1; in createEmbeddingBagByteRowwiseOffsets()2257 auto outTy = getOutputTypeOfFusedRowwiseQuantizedSLS(this, data, segmentDims); in createEmbeddingBagByteRowwiseOffsets()