Searched refs:batchid (Results 1 – 4 of 4) sorted by relevance
35 const int batchid = tx; in magma_interp_3d_device() local37 T *sTmp = swork + batchid * (1 * Q); in magma_interp_3d_device()50 const int batchid = tx / Q; in magma_interp_3d_device() local53 T *sTmp = swork + batchid * (Q * P); // sTmp is input in magma_interp_3d_device()65 const int batchid = tx / Q; in magma_interp_3d_device() local68 T *sTmp = swork + batchid * (Q * Q); in magma_interp_3d_device()
35 const int batchid = tx; in magma_interp_2d_device() local37 T *sTmp = swork + batchid * (1 * Q); in magma_interp_2d_device()50 const int batchid = 0; in magma_interp_2d_device() local52 T *sTmp = swork + batchid * (Q * P); in magma_interp_2d_device()
60 const int batchid = tx;63 T *sTmp = sW1 + batchid * (1 * Q);76 const int batchid = tx / Q;80 T *sTmp = sW1 + batchid * (Q * P); // sTmp is input81 T *sTmp2 = sW2 + batchid * (Q * Q); // sTmp2 is output
57 const int batchid = tx;60 T *sTmp = swork + batchid * (1 * Q);73 const int batchid = 0;76 T *sTmp = swork + batchid * (Q * P);