Searched defs:shared_grad (Results 1 – 2 of 2) sorted by relevance
896 __shared__ T shared_grad[BS][4]; in two_body_ratio_grad_PBC_kernel() local1035 __shared__ T shared_grad[BS][3]; in two_body_ratio_grad_PBC_kernel_fast() local2804 __shared__ T shared_grad[BS][4]; in one_body_ratio_grad_PBC_kernel() local2918 __shared__ T shared_grad[BS][3]; in one_body_ratio_grad_PBC_kernel_fast() local
417 __shared__ T shared_grad[BS][3]; in two_body_ratio_grad_kernel() local1521 __shared__ T shared_grad[BS][3]; in one_body_ratio_grad_kernel() local