Searched refs:check_nested_tensor (Results 1 – 2 of 2) sorted by relevance
42 check_nested_tensor, in use_flash_attention_cpp()
235 inline bool check_nested_tensor(sdp_params const& params, bool debug) { in check_nested_tensor() function