↓ 5 callersFunction_add_conv(
out,
in_channels,
channels,
kernel=1,
stride=1,
pad=0,
num_group=1,
active=T
python/oneflow/test/expensive/pytorch_rexnetv1_lite.py:47
↓ 5 callersFunction_test_bias_add_fused_scale_mask_softmax_dropout(
test_case,
input_shape,
bias_shape,
mask_shape,
input_dtype=flow.float32,
mask_dtype
python/oneflow/test/modules/test_fused_bias_add_scale_mask_softmax_dropout.py:37
↓ 5 callersFunctionnaive_embedding(
x,
cos,
sin,
x_layout,
B,
M,
H,
K,
dims,
merged_dims,
rotary_siz
python/oneflow/test/modules/test_fused_rotary_embedding.py:92