Searched defs:fused_kernel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_transformers.py | 1640 def test_fused_kernels_seq_len_0_inputs(self, device, fused_kernel): argument
1806 fused_kernel, argument
1888 fused_kernel, argument
2552 def test_fused_backwards_throws_determinism_warning(self, device, warn_only, fused_kernel): argument
3001 …def get_dropout_mask(output, fused_kernel, batch_size, n_heads, q_len, kv_len, dropout_p, device=d… argument
3145 def test_fused_kernels_seq_len_1_inputs(self, device, fused_kernel): argument
|
| H A D | test_jit_fuser_te.py | 216 def fused_kernel(a, b): function
|
| /aosp_15_r20/external/gemmlowp/meta/ |
| H A D | base.h | 73 FusedKernel fused_kernel; member
|