Searched defs:num_rhs (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_solve_op_test.py | 161 def _GenerateTestData(self, matrix_shape, num_rhs): argument
|
| D | matrix_solve_ls_op_test.py | 46 def _GenerateTestData(matrix_shape, num_rhs): argument
|
| D | tridiagonal_solve_op_test.py | 670 def _generateData(self, matrix_size, batch_size, num_rhs, seed=42): argument
678 def _generateMatrixData(self, matrix_size, batch_size, num_rhs, seed=42): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal_test.cc | 45 const int64 num_rhs = std::get<2>(spec); in XLA_TEST_P() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 46 const int num_rhs, in SolveForSizeOneOrTwoKernel()
257 const int64 num_rhs = rhs.dim_size(rhs.dims() - 1); in Compute() local
|