| /external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| D | BatchLinearAlgebra.cpp | 284 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
293 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
302 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
311 magma_int_t** ipiv_array, magma_int_t* info_array, magma_int_t batchsize, in magmaLuBatched()
356 magma_int_t* info_array, magma_int_t batchsize, const MAGMAQueue& magma_queue) { in magmaLuNoPivBatched()
364 magma_int_t* info_array, magma_int_t batchsize, const MAGMAQueue& magma_queue) { in magmaLuNoPivBatched()
372 magma_int_t* info_array, magma_int_t batchsize, const MAGMAQueue& magma_queue) { in magmaLuNoPivBatched()
380 magma_int_t* info_array, magma_int_t batchsize, const MAGMAQueue& magma_queue) { in magmaLuNoPivBatched()
428 …double** dB_array, magma_int_t lddb, magma_int_t& info, magma_int_t batchsize, const MAGMAQueue& m… in magmaCholeskySolveBatched()
436 …float** dB_array, magma_int_t lddb, magma_int_t& info, magma_int_t batchsize, const MAGMAQueue& ma… in magmaCholeskySolveBatched()
[all …]
|
| D | BatchLinearAlgebraLib.cpp | 293 int batchsize = calculate_all_batches ? cuda_int_cast(batchCount(A), "batch size") in apply_svd_cusolver_gesvd() local
380 int batchsize = cuda_int_cast(batchCount(A), "batch size"); in apply_svd_cusolver_gesvdj() local
447 int batchsize = cuda_int_cast(batchCount(A), "batch size"); in apply_svd_cusolver_gesvdjBatched() local
1154 auto batchsize = cuda_int_cast(batchCount(self), "batch size"); in apply_orgqr() local
|
| D | CUDASolver.cpp | 1387 int batchsize) { in syevjBatched_bufferSize()
1403 int batchsize) { in syevjBatched_bufferSize()
1419 int batchsize) { in syevjBatched_bufferSize()
1444 int batchsize) { in syevjBatched_bufferSize()
1471 int batchsize) { in syevjBatched()
1489 int batchsize) { in syevjBatched()
1507 int batchsize) { in syevjBatched()
1536 int batchsize) { in syevjBatched()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | UpSampleLinear1d.cu | 35 const int batchsize = idata.size(0); in upsample_linear1d_out_frame() local
82 const int batchsize = idata.size(0); in upsample_linear1d_out_frame_backward() local
|
| D | UpSampleBicubic2d.cu | 33 const int batchsize = idata.size(0); in upsample_bicubic2d_out_frame() local
108 const int batchsize = idata.size(0); in upsample_bicubic2d_backward_out_frame() local
|
| D | UpSampleTrilinear3d.cu | 50 const int batchsize = idata.size(0); in upsample_trilinear3d_out_frame() local
135 const int batchsize = idata.size(0); in upsample_trilinear3d_backward_out_frame() local
|
| D | UpSampleBilinear2d.cu | 44 const int batchsize = idata.size(0); in upsample_bilinear2d_out_frame() local
481 const int batchsize = idata.size(0); in upsample_gen2d_aa_out_frame() local
575 const int batchsize = idata.size(0); in upsample_gen2d_aa_backward_out_frame() local
|
| /external/pytorch/test/onnx/model_defs/ |
| D | word_language_model.py | 23 batchsize=2, argument
|
| /external/pytorch/aten/src/ATen/ |
| D | SparseCsrTensorImpl.cpp | 116 auto batchsize = size.slice(0, batch_dim); in resize_and_clear_() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TriangularOps.cpp | 91 auto batchsize = batchCountTrilTriu(result); in apply_triu_tril() local
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDABlas.cpp | 2096 int n, double** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) { in getrfBatched()
2104 int n, float** dA_array, int ldda, int* ipiv_array, int* info_array, int batchsize) { in getrfBatched()
2117 int batchsize) { in getrfBatched()
2136 int batchsize) { in getrfBatched()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseCsrTensor.cpp | 215 DimVector batchsize = DimVector(size.slice(0, batch_ndim)); in _validate_sparse_compressed_tensor_args_worker() local
|
| /external/rust/android-crates-io/crates/quiche/deps/boringssl/src/tool/ |
| D | speed.cc | 1006 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
|
| /external/cronet/stable/third_party/boringssl/src/tool/ |
| D | speed.cc | 1566 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
|
| /external/boringssl/src/tool/ |
| D | speed.cc | 1496 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
|
| /external/cronet/tot/third_party/boringssl/src/tool/ |
| D | speed.cc | 1566 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
|
| /external/pytorch/test/ |
| D | test_sparse.py | 4269 def _test_meta_sparse_compressed(self, dtype, layout, batchsize, densesize): argument
4328 def _test_print_meta_data(self, dtype, layout, batchsize, sparsesize, densesize): argument
|
| D | test_linalg.py | 603 def cholesky_test_helper(n, batchsize, device, upper): argument
|