Searched defs:with_backward (Results 1 – 2 of 2) sorted by relevance
327 with_backward=False, argument447 def get_flops(q, k, v, backend, with_backward=False): argument536 def get_dense_flops(q, k, v, backend, with_backward=False): argument
9047 def get_refs(with_backward): argument