↓ 59 callersMethodeval(self, batch_size, seq_len, num_heads, head_dim, pe_head_dim)
test/legacy_test/test_fused_partial_rope_op.py:46
↓ 57 callersFunctionto_tensorr""" Constructs a ``paddle.Tensor`` from ``data`` , which can be scalar, tuple, list, numpy\.ndarray, paddle\.Tensor. If the ``data`` is
python/paddle/tensor/creation.py:1102
↓ 55 callersFunctioncreate_test_act_bf16_class(
parent,
atol=1e-2,
grad_check=True,
check_dygraph=True,
check_prim=False,
enable_cin
test/legacy_test/test_activation_op.py:6484
↓ 55 callersFunctioncreate_test_act_fp16_class(
parent,
atol=1e-3,
grad_check=True,
check_dygraph=True,
check_prim=False,
check_prim
test/legacy_test/test_activation_op.py:6285