Searched refs:binary_ufuncs (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_binary_ufuncs.py | 46 binary_ufuncs,
190 filter(lambda op: op.ref is not None and op.ref is not None, binary_ufuncs)
273 @ops(binary_ufuncs)
295 @ops(binary_ufuncs)
317 @ops(binary_ufuncs)
348 @ops(binary_ufuncs)
374 @ops(binary_ufuncs)
399 @ops(binary_ufuncs)
427 @ops(binary_ufuncs)
455 @ops(binary_ufuncs)
[all …]
|
| H A D | test_maskedtensor.py | 19 binary_ufuncs,
850 mt_binary_ufuncs = [op for op in binary_ufuncs if is_binary(op)]
|
| H A D | test_meta.py | 36 binary_ufuncs, op_db, foreach_unary_op_db, foreach_binary_op_db,
1317 @ops(binary_ufuncs, allowed_dtypes=(torch.float32,))
|
| H A D | test_sparse.py | 25 (op_db, reduction_ops, sparse_unary_ufuncs, sparse_masked_reduction_ops, binary_ufuncs)
50 op for op in binary_ufuncs if _op_supports_any_sparse(op) and
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 24413 binary_ufuncs = [op for op in ops_and_refs if isinstance(op, BinaryUfuncInfo)] variable
|