Searched defs:num_rhs (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | tridiagonal_solve_op_gpu.cu.cc | 46 const int num_rhs, in SolveForSizeOneOrTwoKernel()
258 const int64 num_rhs = rhs.dim_size(rhs.dims() - 1); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A D | tridiagonal_solve_op_test.py | 663 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
|
| H A D | matrix_solve_op_test.py | 163 def _GenerateTestData(self, matrix_shape, num_rhs): argument
|
| H A D | matrix_solve_ls_op_test.py | 42 def _GenerateTestData(matrix_shape, num_rhs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | tridiagonal_test.cc | 115 const int64_t num_rhs = std::get<2>(spec); in XLA_TEST_P() local
|