Searched defs:is_transformer_mask (Results 1 – 1 of 1) sorted by relevance
68 …ount, const bool *mask = nullptr, const int head_chunk_size = -1, bool is_transformer_mask = false) in softmax_warp_forward()303 …int batch_count, const bool *mask = nullptr, int chunk_size = -1, bool is_transformer_mask = false) in dispatch_softmax_forward()