| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op.cc | 105 const auto& superdiag = diagonals.row(0); in ComputeMatrix() local
128 SolveWithGaussianEliminationWithPivoting(context, superdiag, diag, in ComputeMatrix()
131 SolveWithThomasAlgorithm(context, superdiag, diag, subdiag, rhs, x); in ComputeMatrix()
139 const MatrixMapRow& superdiag, in SolveWithGaussianEliminationWithPivoting() argument
156 u(0, 1) = superdiag(0); in SolveWithGaussianEliminationWithPivoting()
167 u(i + 1, 1) = superdiag(i + 1); in SolveWithGaussianEliminationWithPivoting()
179 u(i, 2) = superdiag(i + 1); in SolveWithGaussianEliminationWithPivoting()
180 u(i + 1, 1) = -factor * superdiag(i + 1); in SolveWithGaussianEliminationWithPivoting()
195 const MatrixMapRow& superdiag, in SolveWithThomasAlgorithm() argument
209 u(0) = superdiag(0) / diag(0); in SolveWithThomasAlgorithm()
[all …]
|
| D | tridiagonal_matmul_op_gpu.cu.cc | 38 const Scalar* __restrict__ superdiag, in TridiagonalMatMulKernel() argument
50 result = result + superdiag[row_id] * rhs[i + n]; in TridiagonalMatMulKernel()
63 const Tensor& superdiag = context->input(0); in Compute() local
84 0, device.stream(), batch_size, m, n, superdiag.flat<Scalar>().data(), in Compute()
|
| D | tridiagonal_solve_op_gpu.cu.cc | 139 const auto& superdiag = diagonals.row(0); in ComputeMatrix() local
164 SolveWithGtsv(context, cusparse_solver, superdiag.data(), diag.data(), in ComputeMatrix()
178 SolveWithGtsv(context, cusparse_solver, superdiag.data(), diag.data(), in ComputeMatrix()
200 const Scalar* superdiag, const Scalar* diag, in SolveWithGtsv() argument
208 num_eqs, num_rhs, subdiag, diag, superdiag, rhs, in SolveWithGtsv()
219 num_eqs, num_rhs, subdiag, diag, superdiag, rhs, in SolveWithGtsv()
298 const Scalar* superdiag = lhs_data; in ComputeWithGtsvBatched() local
316 matrix_size, subdiag, diag, superdiag, x, in ComputeWithGtsvBatched()
324 matrix_size, subdiag, diag, superdiag, x, in ComputeWithGtsvBatched()
|
| D | tridiagonal_matmul_op.cc | 87 const auto& superdiag = inputs[0].row(0); in ComputeMatrix() local
105 ConstVectorMap superdiag_map(superdiag.data(), m - 1); in ComputeMatrix()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_tridiag_test.py | 52 superdiag = math_ops.conj(subdiag)
53 superdiag = manip_ops.roll(superdiag, shift=-1, axis=-1)
55 superdiag = linear_operator_test_util.random_sign_uniform(
59 [superdiag, diag, subdiag], axis=-2)
69 diagonals = [superdiag, diag, subdiag]
71 diagonals = array_ops.stack([superdiag, diag, subdiag], axis=-2)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_matmul_op_test.py | 42 superdiag, argument
48 superdiag_extended = np.pad(superdiag, [0, 1], 'constant')
51 diags_matrix = np.diag(superdiag, 1) + np.diag(maindiag, 0) + np.diag(
94 def _makeTridiagonalMatrix(self, superdiag, maindiag, subdiag): argument
98 super_part = array_ops.pad(array_ops.matrix_diag(superdiag), super_pad)
145 superdiag = self._randomComplexArray((b, m - 1))
149 matrix = np.stack([np.diag(superdiag[i], 1) + \
|
| D | tridiagonal_solve_op_test.py | 642 superdiag = array_ops.placeholder(dtypes.float64, shape=[None])
647 x = linalg_impl.tridiagonal_solve((superdiag, diag, subdiag),
657 superdiag: [2, 1, 4, 20],
684 superdiag = 2 * np.abs(np.random.randn(matrix_size - 1))
685 matrix = sparse.diags([superdiag, diag, subdiag], [1, 0, -1]).toarray()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 555 superdiag, maindiag, subdiag = diagonals
557 not superdiag.shape[:-1].is_compatible_with(maindiag.shape[:-1])):
561 subdiag.shape, maindiag.shape, superdiag.shape))
577 superdiag = pad_if_necessary(superdiag, 'superdiagonal', [0, 1])
579 diagonals = array_ops.stack((superdiag, maindiag, subdiag), axis=-2)
700 superdiag = diagonals[..., 0, :]
704 superdiag, maindiag, subdiag = diagonals
714 superdiag = diags[..., 0, :]
722 superdiag = array_ops.expand_dims(superdiag, -2)
726 return linalg_ops.tridiagonal_mat_mul(superdiag, maindiag, subdiag, rhs, name)
|
| D | linear_operator_tridiag.py | 291 superdiag, diag, subdiag = array_ops.unstack(
296 new_subdiag = manip_ops.roll(superdiag, shift=1, axis=-1)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 285 ShapeHandle superdiag; in TridiagonalMatMulShapeFn() local
291 TF_RETURN_IF_ERROR(c->WithRankAtLeast(c->input(0), 2, &superdiag)); in TridiagonalMatMulShapeFn()
301 TF_RETURN_IF_ERROR(c->Subshape(superdiag, 0, -2, &superdiag_batch_shape)); in TridiagonalMatMulShapeFn()
305 TF_RETURN_IF_ERROR(c->Merge(superdiag, maindiag, &superdiag)); in TridiagonalMatMulShapeFn()
312 TF_RETURN_IF_ERROR(c->Merge(superdiag, maindiag, &maindiag)); in TridiagonalMatMulShapeFn()
|
| D | ops.pbtxt | 56777 name: "superdiag"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TridiagonalMatMul.pbtxt | 4 name: "superdiag"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TridiagonalMatMul.pbtxt | 4 name: "superdiag"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TridiagonalMatMul.pbtxt | 5 name: "superdiag"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 1014 superdiag = array_ops.concat((diags[..., 2, 1:], zeros), axis=-1)
1021 superdiag = array_ops.pad(diags[..., 2, 1:], superdiag_pad)
1025 return array_ops.stack([superdiag, diag, subdiag], axis=-2)
1049 superdiag = math_ops.reduce_sum(
1058 superdiag = math_ops.reduce_sum(
1064 return array_ops.stack([superdiag, diag, subdiag], axis=-2)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 5157 …argspec: "args=[\'superdiag\', \'maindiag\', \'subdiag\', \'rhs\', \'name\'], varargs=None, keywor…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 5157 …argspec: "args=[\'superdiag\', \'maindiag\', \'subdiag\', \'rhs\', \'name\'], varargs=None, keywor…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 15036 func TridiagonalMatMul(scope *Scope, superdiag tf.Output, maindiag tf.Output, subdiag tf.Output, rh…
15043 superdiag, maindiag, subdiag, rhs,
|