Searched refs:check_cudnn_tensor_shapes (Results 1 – 1 of 1) sorted by relevance
313 bool check_cudnn_tensor_shapes(sdp_params const& params, bool debug) { in check_cudnn_tensor_shapes() function539 check_cudnn_tensor_shapes, in can_use_cudnn_attention()