Searched defs:num_splits_key (Results 1 – 2 of 2) sorted by relevance
248 _efficient_attention_backward( in _efficient_attention_backward()
713 int16_t num_splits_key = 1; // We use `gridDim.x` inside kernel member