| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_tridiag.py | 129 diagonals, argument
175 diagonals=diagonals,
184 with ops.name_scope(name, values=[diagonals]):
191 d, name='diag_{}'.format(i)) for i, d in enumerate(diagonals)]
195 diagonals, name='diagonals')
210 return self.diagonals.shape
213 d_shape = self.diagonals.shape[:-2].concatenate(
214 self.diagonals.shape[-1])
217 self.diagonals[0].shape[:-1],
218 self.diagonals[1].shape[:-1])
[all …]
|
| D | linalg_impl.py | 448 def tridiagonal_solve(diagonals, argument
547 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
552 if not isinstance(diagonals, (tuple, list)) or len(diagonals) != 3:
555 superdiag, maindiag, subdiag = diagonals
579 diagonals = array_ops.stack((superdiag, maindiag, subdiag), axis=-2)
580 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
585 m1 = tensor_shape.dimension_value(diagonals.shape[-1])
586 m2 = tensor_shape.dimension_value(diagonals.shape[-2])
592 diagonals = array_ops.matrix_diag_part(
593 diagonals, k=(-1, 1), padding_value=0., align='LEFT_RIGHT')
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_tridiag_test.py | 69 diagonals = [superdiag, diag, subdiag]
71 diagonals = array_ops.stack([superdiag, diag, subdiag], axis=-2)
73 diagonals = matrix
75 lin_op_diagonals = diagonals
86 diagonals=lin_op_diagonals,
140 diagonals = [
145 diagonals, diagonals_format='sequence')
150 diagonals = [
156 diagonals, diagonals_format='sequence')
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixSetDiagV3.pbtxt | 32 Some diagonals are shorter than `max_diag_len` and need to be padded. `align` is
44 same shape and values as `input`, except for the specified diagonals of the
50 `num_diags` is the number of diagonals, `num_diags = k[1] - k[0] + 1`.
112 # A band of diagonals.
113 diagonals = np.array([[[0, 9, 1], # Diagonal shape: (2, 4, 3)
121 tf.matrix_set_diag(input, diagonals, k = (-1, 2))
130 diagonals = np.array([[[9, 1, 0], # Diagonal shape: (2, 4, 3)
138 tf.matrix_set_diag(input, diagonals, k = (-1, 2), align="LEFT_RIGHT")
|
| D | api_def_MatrixSetDiagV2.pbtxt | 32 same shape and values as `input`, except for the specified diagonals of the
38 `num_diags` is the number of diagonals, `num_diags = k[1] - k[0] + 1`.
88 # A band of diagonals.
89 diagonals = np.array([[[1, 2, 3], # Diagonal shape: (2, 2, 3)
93 tf.matrix_set_diag(diagonals, k = (-1, 0))
|
| D | api_def_MatrixDiagV3.pbtxt | 48 Some diagonals are shorter than `max_diag_len` and need to be padded. `align` is
61 diagonals of a matrix, with everything else padded with `padding`. `num_rows`
83 Otherwise, `M` is treated as the number of diagonals for the matrix in the
133 diagonals = np.array([[[0, 8, 9], # Input shape: (2, 2, 3)
139 tf.matrix_diag(diagonals, k = (-1, 1))
148 diagonals = np.array([[[8, 9, 0], # Input shape: (2, 2, 3)
154 tf.matrix_diag(diagonals, k = (-1, 1), align="LEFT_RIGHT")
|
| D | api_def_MatrixDiagV2.pbtxt | 49 diagonals of a matrix, with everything else padded with `padding`. `num_rows`
71 Otherwise, `M` is treated as the number of diagonals for the matrix in the
109 # A band of diagonals.
110 diagonals = np.array([[[1, 2, 3], # Input shape: (2, 2, 3)
114 tf.matrix_diag(diagonals, k = (-1, 0))
|
| D | api_def_MatrixDiagPartV3.pbtxt | 30 Some diagonals are shorter than `max_diag_len` and need to be padded. `align` is
41 Returns a tensor with the `k[0]`-th to `k[1]`-th diagonals of the batched
45 Let `max_diag_len` be the maximum length among all diagonals to be extracted,
47 Let `num_diags` be the number of diagonals to extract,
|
| D | api_def_MatrixDiagPartV2.pbtxt | 29 Returns a tensor with the `k[0]`-th to `k[1]`-th diagonals of the batched
33 Let `max_diag_len` be the maximum length among all diagonals to be extracted,
35 Let `num_diags` be the number of diagonals to extract,
|
| D | api_def_TridiagonalSolve.pbtxt | 5 name: "diagonals"
8 tridiagonal matrices with three rows being the superdiagonal, diagonals, and
|
| D | api_def_TridiagonalMatMul.pbtxt | 14 Tensor of shape `[..., 1, M]`, representing main diagonals of tri-diagonal
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 289 StatusOr<XlaOp> ThomasSolver(XlaOp diagonals, XlaOp rhs) { in ThomasSolver() argument
290 XlaBuilder* builder = diagonals.builder(); in ThomasSolver()
291 TF_ASSIGN_OR_RETURN(Shape diagonals_shape, builder->GetShape(diagonals)); in ThomasSolver()
295 SliceInDim(diagonals, /*start_index=*/0, /*limit_index=*/1, in ThomasSolver()
298 SliceInDim(diagonals, /*start_index=*/1, /*limit_index=*/2, in ThomasSolver()
301 SliceInDim(diagonals, /*start_index=*/2, /*limit_index=*/3, in ThomasSolver()
|
| D | tridiagonal.h | 28 StatusOr<XlaOp> ThomasSolver(XlaOp diagonals, XlaOp rhs);
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 137 const auto diagonals = inputs[0]; in ComputeMatrix() local
139 const auto& superdiag = diagonals.row(0); in ComputeMatrix()
141 const auto& diag = diagonals.row(1); in ComputeMatrix()
143 const auto& subdiag = diagonals.row(2); in ComputeMatrix()
155 SolveForSizeOneOrTwo(context, diagonals.data(), rhs.data(), x.data(), m, in ComputeMatrix()
223 void SolveForSizeOneOrTwo(OpKernelContext* context, const Scalar* diagonals, in SolveForSizeOneOrTwo() argument
231 device.stream(), m, diagonals, rhs, k, output, in SolveForSizeOneOrTwo()
|
| D | tridiagonal_solve_op.cc | 102 const auto diagonals = inputs[0]; in ComputeMatrix() local
105 const auto& superdiag = diagonals.row(0); in ComputeMatrix()
107 const auto& diag = diagonals.row(1); in ComputeMatrix()
109 const auto& subdiag = diagonals.row(2); in ComputeMatrix()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tridiagonal_ops.cc | 31 auto diagonals = ctx->Input(0); in Compile() local
42 auto result = xla::tridiagonal::ThomasSolver(diagonals, rhs); in Compile()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TridiagonalSolve.pbtxt | 4 name: "diagonals"
31 name: "diagonals"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TridiagonalSolve.pbtxt | 4 name: "diagonals"
31 name: "diagonals"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.-linear-operator-tridiag.pbtxt | 18 name: "diagonals"
95 …argspec: "args=[\'self\', \'diagonals\', \'diagonals_format\', \'is_non_singular\', \'is_self_adjo…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.-linear-operator-tridiag.pbtxt | 18 name: "diagonals"
95 …argspec: "args=[\'self\', \'diagonals\', \'diagonals_format\', \'is_non_singular\', \'is_self_adjo…
|
| D | tensorflow.linalg.pbtxt | 261 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'name\'], varargs=None, keywords=No…
265 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'transpose_rhs\', \'conjugate_rhs\'…
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_solve_ops_test.py | 488 diagonals = tuple(_tf_ones(shape) for shape in diags_tuple_shapes)
489 self._assertRaises(diagonals, _tf_ones(rhs_shape), "sequence")
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/core/ |
| D | tessellator.hpp | 320 …void StitchRegular(bool bTrapezoid, DIAGONALS diagonals, int baseIndexOffset, int numInsideEdgePoi…
|
| /external/mesa3d/src/gallium/auxiliary/tessellator/ |
| D | tessellator.hpp | 320 …void StitchRegular(bool bTrapezoid, DIAGONALS diagonals, int baseIndexOffset, int numInsideEdgePoi…
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_solve_op_test.py | 528 diagonals = tuple(_tf_ones(shape) for shape in diags_tuple_shapes)
529 self._assertRaises(diagonals, _tf_ones(rhs_shape), "sequence")
|