Rate this Page
★ ★ ★ ★ ★

torch.utils.flop_counter.sdpa_backward_flop_count#

torch.utils.flop_counter.sdpa_backward_flop_count(grad_out_shape, query_shape, key_shape, value_shape)[source]#
On this page
Show Source
PyTorch Libraries