↓ 11 callersFunction_autograd_grad(
outputs, inputs, grad_outputs=None, create_graph=False, retain_graph=None,
)
python/oneflow/autograd/functional.py:175
↓ 10 callersFunction_test_fused_multi_head_attention_inference_with_attn_bias(
test_case,
batch_size,
num_heads,
query_seq_len,
kv_seq_len,
query_head_size,
va
python/oneflow/test/modules/test_fused_attention_ops.py:357