↓ 8 callersFunctioncompare(a, b, a_name, b_name, atol=1e-5, rtol=1e-8)
python/oneflow/test/modules/test_fused_bias_add_scale_mask_softmax_dropout.py:94
↓ 7 callersMethod__init__(
self,
img_size=224,
patch_size=16,
in_chans=3,
num_classes=1000,
python/oneflow/test/expensive/pytorch_levit.py:377
↓ 7 callersFunction_fused_mha(
query,
key,
value,
num_heads,
attn_mask_type="none",
attn_bias=None,
causal_diag
python/oneflow/test/modules/test_fused_attention_ops.py:157
↓ 7 callersFunction_ref(
query,
key,
value,
num_heads,
attn_mask_type="none",
attn_bias=None,
causal_diag
python/oneflow/test/modules/test_fused_attention_ops.py:27
↓ 7 callersFunction_test_conv2d(
test_case, conv, data, weight, output, bias=None, device="cuda",
)
python/oneflow/test/modules/test_conv2d.py:1154