↓ 31 callersMethodtest_binary_body(
self,
x_shape,
y_shape,
out_shape,
x_placements,
y_placement
test/auto_parallel/semi_auto_parallel_for_elementwise.py:57
↓ 29 callersFunction_math_attention(
query: Tensor,
key: Tensor,
value: Tensor,
mask: Tensor,
dropout_rate: float = ...,
python/paddle/nn/functional/flash_attention.py:76
↓ 29 callersFunctioncreate_test_case(
baseclass,
api,
support_int_types=["uint8", "int8", "int16", "int32", "int64"],
**kwargs,
)
test/legacy_test/test_tensor_type_autocast.py:32