Home
last modified time | relevance | path

Searched defs:batchsize (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebra.cpp284 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 …]
H A DCUDASolver.cpp1387 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()
H A DBatchLinearAlgebraLib.cpp293 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleLinear1d.cu35 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
H A DUpSampleBicubic2d.cu33 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
H A DUpSampleTrilinear3d.cu50 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
H A DUpSampleBilinear2d.cu44 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
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp2096 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()
/aosp_15_r20/external/pytorch/test/onnx/model_defs/
H A Dword_language_model.py23 batchsize=2, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorImpl.cpp116 auto batchsize = size.slice(0, batch_dim); in resize_and_clear_() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTriangularOps.cpp91 auto batchsize = batchCountTrilTriu(result); in apply_triu_tril() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp215 DimVector batchsize = DimVector(size.slice(0, batch_ndim)); in _validate_sparse_compressed_tensor_args_worker() local
/aosp_15_r20/external/rust/android-crates-io/crates/quiche/deps/boringssl/src/tool/
Dspeed.cc1006 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
/aosp_15_r20/external/rust/crates/quiche/deps/boringssl/src/tool/
Dspeed.cc1006 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
/aosp_15_r20/external/cronet/third_party/boringssl/src/tool/
H A Dspeed.cc1301 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py4269 def _test_meta_sparse_compressed(self, dtype, layout, batchsize, densesize): argument
4328 def _test_print_meta_data(self, dtype, layout, batchsize, sparsesize, densesize): argument
H A Dtest_linalg.py603 def cholesky_test_helper(n, batchsize, device, upper): argument
/aosp_15_r20/external/boringssl/src/tool/
H A Dspeed.cc1413 size_t batchsize, const std::string &selected) { in SpeedTrustToken()
/aosp_15_r20/prebuilts/devtools/tools/lib/
HDjython-standalone-2.5.3.jarMETA-INF/MANIFEST.MF META-INF/LICENSE META-INF/NOTICE META- ...
/aosp_15_r20/prebuilts/jdk/jdk8/linux-x86/lib/
Dct.symMETA-INF/ META-INF/MANIFEST.MF META-INF/sym/rt. ...
/aosp_15_r20/prebuilts/jdk/jdk8/darwin-x86/lib/
Dct.symMETA-INF/ META-INF/MANIFEST.MF META-INF/sym/rt. ...
/aosp_15_r20/prebuilts/jdk/jdk8/linux-x86/jre/lib/
Drt.jarMETA-INF/ META-INF/MANIFEST.MF com/oracle/net/Sdp$ ...
/aosp_15_r20/prebuilts/jdk/jdk8/darwin-x86/jre/lib/
Drt.jarMETA-INF/ META-INF/MANIFEST.MF apple/applescript/AppleScriptEngine.class AppleScriptEngine ...