Searched defs:d_v (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | MHA.cpp | 17 int64_t d_v, in run_cudnn_SDP_fprop()
39 int64_t d_v, in run_cudnn_SDP_bprop()
134 int64_t d_v; member
147 int64_t d_v, in setMHAParams()
260 int64_t d_v, in build_graph_and_tensors()
395 int64_t d_v, in build_graph_and_tensors_backward()
527 int64_t d_v, in run_cudnn_SDP_fprop()
615 int64_t d_v, in run_cudnn_SDP_bprop()
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_cuda.cpp | 1213 PaddedBuffer<float> d_v(B_SIZE); in TEST() local
1310 PaddedBuffer<float> d_v(B_SIZE); in TEST() local
1408 PaddedBuffer<float> d_v(B_SIZE); in TEST() local
1506 PaddedBuffer<float> d_v(B_SIZE); in TEST() local
1609 PaddedBuffer<float> d_v(OUTER_SIZE, B_SIZE); in TEST() local
1739 PaddedBuffer<float> d_v(OUTER_EXTENT, B_EXTENT); in TEST() local
1880 PaddedBuffer<float> d_v(OUTER_SIZE, B_SIZE); in TEST() local
2018 PaddedBuffer<float> d_v(OUTER_SIZE, B_SIZE); in TEST() local
2141 PaddedBuffer<float> d_v(OUTER_SIZE, B_SIZE); in TEST() local
2268 PaddedBuffer<float> d_v(OUTER_B_SIZE, B_SIZE); in TEST() local
|
| H A D | test_aten.cpp | 116 PaddedBuffer<int> d_v(kTotalSize); in TEST() local
152 PaddedBuffer<float> d_v(kTotalSize); in TEST() local
188 PaddedBuffer<int> d_v(kTotalSize); in TEST() local
224 PaddedBuffer<float> d_v(kTotalSize); in TEST() local
260 PaddedBuffer<float> d_v(kTotalSize); in TEST() local
298 PaddedBuffer<int> d_v(kTotalSize); in TEST() local
339 PaddedBuffer<float> d_v(kTotalSize); in TEST() local
|
| H A D | test_reductions.cpp | 1309 PaddedBuffer<float> d_v(L, "d"); in TEST() local
1383 PaddedBuffer<float> d_v(L, "d"); in TEST() local
1457 PaddedBuffer<float> d_v(L, "d"); in TEST() local
|
| H A D | test_loopnest.cpp | 977 PaddedBuffer<float> d_v(M, N, K); in TEST() local
1042 PaddedBuffer<float> d_v(M, K); in TEST() local
1133 PaddedBuffer<float> d_v(M, K); in InlineFunc01Helper() local
2792 PaddedBuffer<float> d_v(M, K); in TEST() local
|
| /aosp_15_r20/external/igt-gpu-tools/tests/i915/ |
| H A D | gem_exec_balancer.c | 406 uint64_t d_t, d_v; in measure_load() local
426 uint64_t d_t, d_v[num]; in measure_min_load() local
460 uint64_t d_t, d_v[num]; in measure_all_load() local
|
| /aosp_15_r20/external/clang/test/CodeGenCUDA/ |
| H A D | device-var-init.cu | 16 __device__ int d_v; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/ |
| H A D | attention.cu | 120 for (int32_t d_v = threadIdx.x; d_v < D / VEC; d_v += blockDim.x) { in transform_bias_rescale_qkv_kernel() local
227 for (int32_t d_v = threadIdx.x; d_v < D / VEC; d_v += blockDim.x) { in transform_bias_rescale_qkv_add_padding_kernel() local
|