Searched defs:acc_o_rowcol (Results 1 – 1 of 1) sorted by relevance
147 …Tensor acc_o_rowcol = make_tensor(acc_o.data(), pytorch_flash::convert_layout_acc_rowcol(acc_o.lay… in softmax_rescale_o() local171 …Tensor acc_o_rowcol = make_tensor(acc_o.data(), pytorch_flash::convert_layout_acc_rowcol(acc_o.lay… member