Home
last modified time | relevance | path

Searched refs:batchCount (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_sparse.cc287 Scalar* x, int batchCount, in Gtsv2StridedBatchImpl() argument
291 AsCudaComplex(du), AsCudaComplex(x), batchCount, batchStride, pBuffer)); in Gtsv2StridedBatchImpl()
299 int batchCount, int batchStride, void* pBuffer) const { \
303 batchCount, batchStride, pBuffer); \
311 const Scalar* d, const Scalar* du, const Scalar* x, int batchCount, in Gtsv2StridedBatchBufferSizeImpl() argument
315 AsCudaComplex(x), batchCount, batchStride, in Gtsv2StridedBatchBufferSizeImpl()
324 const Scalar* x, int batchCount, int batchStride, \
329 *gpusparse_handle_, m, dl, d, du, x, batchCount, batchStride, \
Dcuda_sparse.h240 const Scalar* du, Scalar* x, int batchCount,
249 const Scalar* x, int batchCount,
/external/swiftshader/src/Pipeline/
DComputeProgram.cpp241 const uint32_t batchCount = 16; in run() local
245 for(uint32_t batchID = 0; batchID < batchCount && batchID < groupCount; batchID++) in run()
252 for(uint32_t groupIndex = batchID; groupIndex < groupCount; groupIndex += batchCount) in run()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcusparse_11_0.inc3272 const float *du, const float *x, int batchCount, int batchStride,
3280 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3286 const double *du, const double *x, int batchCount, int batchStride,
3294 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3300 const cuComplex *du, const cuComplex *x, int batchCount, int batchStride,
3308 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3315 const cuDoubleComplex *x, int batchCount, int batchStride,
3323 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3329 const float *du, float *x, int batchCount, int batchStride, void *pBuffer) {
3335 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride, pBuffer);
[all …]
Dcusparse_10_1.inc4327 const float *du, float *x, int batchCount, int batchStride) {
4333 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4338 const double *du, double *x, int batchCount, int batchStride) {
4344 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4349 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4355 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4361 int batchCount, int batchStride) {
4367 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4372 const float *du, const float *x, int batchCount, int batchStride,
4380 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]
Dcusparse_10_2.inc4327 const float *du, float *x, int batchCount, int batchStride) {
4333 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4338 const double *du, double *x, int batchCount, int batchStride) {
4344 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4349 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4355 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4361 int batchCount, int batchStride) {
4367 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4372 const float *du, const float *x, int batchCount, int batchStride,
4380 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]
Dcusparse_10_0.inc4356 const float *du, float *x, int batchCount, int batchStride) {
4362 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4367 const double *du, double *x, int batchCount, int batchStride) {
4373 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4378 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4384 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4390 int batchCount, int batchStride) {
4396 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4401 const float *du, const float *x, int batchCount, int batchStride,
4409 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]
Dcublas_11_0.inc2708 float *const Carray[], int ldc, int batchCount) {
2716 ldb, beta, Carray, ldc, batchCount);
2724 double *const Carray[], int ldc, int batchCount) {
2732 ldb, beta, Carray, ldc, batchCount);
2740 cuComplex *const Carray[], int ldc, int batchCount) {
2749 ldb, beta, Carray, ldc, batchCount);
2757 cuComplex *const Carray[], int ldc, int batchCount) {
2766 ldb, beta, Carray, ldc, batchCount);
2776 cuDoubleComplex *const Carray[], int ldc, int batchCount) {
2785 ldb, beta, Carray, ldc, batchCount);
[all …]
Dcublas_10_1.inc2708 float *const Carray[], int ldc, int batchCount) {
2716 ldb, beta, Carray, ldc, batchCount);
2724 double *const Carray[], int ldc, int batchCount) {
2732 ldb, beta, Carray, ldc, batchCount);
2740 cuComplex *const Carray[], int ldc, int batchCount) {
2749 ldb, beta, Carray, ldc, batchCount);
2757 cuComplex *const Carray[], int ldc, int batchCount) {
2766 ldb, beta, Carray, ldc, batchCount);
2776 cuDoubleComplex *const Carray[], int ldc, int batchCount) {
2785 ldb, beta, Carray, ldc, batchCount);
[all …]
Dcublas_10_2.inc2708 float *const Carray[], int ldc, int batchCount) {
2716 ldb, beta, Carray, ldc, batchCount);
2724 double *const Carray[], int ldc, int batchCount) {
2732 ldb, beta, Carray, ldc, batchCount);
2740 cuComplex *const Carray[], int ldc, int batchCount) {
2749 ldb, beta, Carray, ldc, batchCount);
2757 cuComplex *const Carray[], int ldc, int batchCount) {
2766 ldb, beta, Carray, ldc, batchCount);
2776 cuDoubleComplex *const Carray[], int ldc, int batchCount) {
2785 ldb, beta, Carray, ldc, batchCount);
[all …]
Dcublas_10_0.inc2583 float *const Carray[], int ldc, int batchCount) {
2591 ldb, beta, Carray, ldc, batchCount);
2599 double *const Carray[], int ldc, int batchCount) {
2607 ldb, beta, Carray, ldc, batchCount);
2615 cuComplex *const Carray[], int ldc, int batchCount) {
2624 ldb, beta, Carray, ldc, batchCount);
2632 cuComplex *const Carray[], int ldc, int batchCount) {
2641 ldb, beta, Carray, ldc, batchCount);
2651 cuDoubleComplex *const Carray[], int ldc, int batchCount) {
2660 ldb, beta, Carray, ldc, batchCount);
[all …]
Dcublas_9_0.inc2558 float *Carray[], int ldc, int batchCount) {
2566 ldb, beta, Carray, ldc, batchCount);
2574 double *Carray[], int ldc, int batchCount) {
2582 ldb, beta, Carray, ldc, batchCount);
2590 cuComplex *Carray[], int ldc, int batchCount) {
2598 ldb, beta, Carray, ldc, batchCount);
2606 cuComplex *Carray[], int ldc, int batchCount) {
2614 ldb, beta, Carray, ldc, batchCount);
2623 cuDoubleComplex *Carray[], int ldc, int batchCount) {
2632 ldb, beta, Carray, ldc, batchCount);
[all …]
Dcusparse_9_0.inc4103 const float *du, float *x, int batchCount, int batchStride) {
4109 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4114 const double *du, double *x, int batchCount, int batchStride) {
4120 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4125 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4131 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4137 int batchCount, int batchStride) {
4143 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4148 const float *du, const float *x, int batchCount, int batchStride,
4156 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]