Home
last modified time | relevance | path

Searched defs:matrix_size (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/libopus/tests/
H A Dtest_opus_projection.c211 opus_int32 matrix_size; in test_creation_arguments() local
300 opus_int32 matrix_size = 0; in test_encode_decode() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_solve_op_gpu.cu.cc259 const int64 matrix_size = lhs.dim_size(ndims - 1); in Compute() local
297 int matrix_size = lhs.dim_size(ndims - 1); in ComputeWithGtsvBatched() local
H A Ddeterminant_op_gpu.cu.cc57 const int matrix_size = n * n; in DeterminantFromPivotedLUKernel() local
H A Dbanded_triangular_solve_op.cc69 int matrix_size = in_x.dim_size(2); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dtridiagonal_solve_op_test.py663 def _generateData(self, matrix_size, batch_size, num_rhs, seed=42): argument
671 def _generateMatrixData(self, matrix_size, batch_size, num_rhs, seed=42): argument
/aosp_15_r20/external/libtextclassifier/native/utils/tflite/
H A Ddist_diversification.cc32 const int matrix_size, in DiversifyByDistance()
/aosp_15_r20/external/libopus/src/
H A Dopus_projection_decoder.c110 opus_int32 matrix_size; in opus_projection_decoder_get_size() local
/aosp_15_r20/external/webrtc/modules/audio_processing/utility/
H A Ddelay_estimator.cc73 int matrix_size, in BitCountComparison()
/aosp_15_r20/external/pytorch/test/xpu/
H A Dtest_gemm.py1063 def gen_sizes_matmul(self, x_dim, y_dim=4, matrix_size=4, batch_size=3): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py4416 def random_hermitian_psd_matrix(matrix_size, *batch_dims, dtype=torch.double, device='cpu'): argument
4429 def random_symmetric_pd_matrix(matrix_size, *batch_dims, **kwargs): argument
4446 def random_hermitian_pd_matrix(matrix_size, *batch_dims, dtype, device): argument
4556 def random_sparse_pd_matrix(matrix_size, density=0.01, **kwargs): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_linalg.py2442 def run_subtest(actual_rank, matrix_size, batches, device, svd_lowrank, **options): argument
4433 def gen_sizes_matmul(self, x_dim, y_dim=4, matrix_size=4, batch_size=3): argument
8105 def run_subtest(guess_rank, actual_rank, matrix_size, batches, device, pca, **options): argument