Searched defs:grad_out_shape (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/utils/ |
| D | flop_counter.py | 139 grad_out_shape, argument 436 def sdpa_backward_flop_count(grad_out_shape, query_shape, key_shape, value_shape): argument 464 def sdpa_backward_flop(grad_out_shape, query_shape, key_shape, value_shape, *args, out_shape=None, … argument
|