Searched refs:sparse_mask_like_prepare_sparse_inputs (Results 1 – 1 of 1) sorted by relevance
717 static std::tuple<Tensor, Tensor, OptTensor> sparse_mask_like_prepare_sparse_inputs( in sparse_mask_like_prepare_sparse_inputs() function804 auto [lhs, rhs, lhs_hash_opt] = sparse_mask_like_prepare_sparse_inputs("sparse_mask", t, mask); in sparse_mask()835 …auto [lhs, rhs, lhs_hash_opt] = sparse_mask_like_prepare_sparse_inputs("_sparse_mask_projection", … in sparse_mask_projection()