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.h244 const Scalar* du, Scalar* x, int batchCount,
253 const Scalar* x, int batchCount,
/external/swiftshader/src/Pipeline/
DComputeProgram.cpp221 const uint32_t batchCount = 16; in run() local
225 for(uint32_t batchID = 0; batchID < batchCount && batchID < groupCount; batchID++) in run()
232 for(uint32_t groupIndex = batchID; groupIndex < groupCount; groupIndex += batchCount) in run()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcusparse_11_0.inc3347 const float *du, const float *x, int batchCount, int batchStride,
3355 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3361 const double *du, const double *x, int batchCount, int batchStride,
3369 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3375 const cuComplex *du, const cuComplex *x, int batchCount, int batchStride,
3383 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3390 const cuDoubleComplex *x, int batchCount, int batchStride,
3398 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
3404 const float *du, float *x, int batchCount, int batchStride, void *pBuffer) {
3410 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride, pBuffer);
[all …]
Dcublas_11_0.inc2013 float *const yarray[], int incy, int batchCount) {
2021 yarray, incy, batchCount);
2029 double *const yarray[], int incy, int batchCount) {
2037 yarray, incy, batchCount);
2045 cuComplex *const yarray[], int incy, int batchCount) {
2053 yarray, incy, batchCount);
2062 cuDoubleComplex *const yarray[], int incy, int batchCount) {
2070 yarray, incy, batchCount);
2079 float *y, int incy, long long int stridey, int batchCount) {
2087 beta, y, incy, stridey, batchCount);
[all …]
Dcusparse_10_1.inc4359 const float *du, float *x, int batchCount, int batchStride) {
4365 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4370 const double *du, double *x, int batchCount, int batchStride) {
4376 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4381 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4387 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4393 int batchCount, int batchStride) {
4399 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4404 const float *du, const float *x, int batchCount, int batchStride,
4412 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]
Dcusparse_10_2.inc4359 const float *du, float *x, int batchCount, int batchStride) {
4365 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4370 const double *du, double *x, int batchCount, int batchStride) {
4376 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4381 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4387 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4393 int batchCount, int batchStride) {
4399 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4404 const float *du, const float *x, int batchCount, int batchStride,
4412 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]
Dcusparse_10_0.inc4388 const float *du, float *x, int batchCount, int batchStride) {
4394 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4399 const double *du, double *x, int batchCount, int batchStride) {
4405 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4410 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4416 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4422 int batchCount, int batchStride) {
4428 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4433 const float *du, const float *x, int batchCount, int batchStride,
4441 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[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_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_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.inc4135 const float *du, float *x, int batchCount, int batchStride) {
4141 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4146 const double *du, double *x, int batchCount, int batchStride) {
4152 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4157 const cuComplex *du, cuComplex *x, int batchCount, int batchStride) {
4163 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4169 int batchCount, int batchStride) {
4175 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride);
4180 const float *du, const float *x, int batchCount, int batchStride,
4188 return func_ptr(handle, m, dl, d, du, x, batchCount, batchStride,
[all …]