| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixDiagV3.pbtxt | 110 tf.matrix_diag(diagonal) ==> [[[1, 0, 0, 0], # Output shape: (2, 4, 4)
122 tf.matrix_diag(diagonal, k = 1)
139 tf.matrix_diag(diagonals, k = (-1, 1))
154 tf.matrix_diag(diagonals, k = (-1, 1), align="LEFT_RIGHT")
164 tf.matrix_diag(diagonal, k = -1, num_rows = 3, num_cols = 4)
170 tf.matrix_diag(diagonal, k = -1, num_rows = 3, padding_value = 9)
|
| D | api_def_MatrixDiagV2.pbtxt | 87 tf.matrix_diag(diagonal) ==> [[[1, 0, 0, 0], # Output shape: (2, 4, 4)
99 tf.matrix_diag(diagonal, k = 1)
114 tf.matrix_diag(diagonals, k = (-1, 0))
124 tf.matrix_diag(diagonal, k = -1, num_rows = 3, num_cols = 4)
130 tf.matrix_diag(diagonal, k = -1, num_rows = 3, padding_value = 9)
|
| D | api_def_MatrixDiag.pbtxt | 32 tf.matrix_diag(diagonal) ==> [[[1, 0, 0, 0]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | matrix_diag.cc | 27 namespace matrix_diag { namespace
134 static TfLiteRegistration r = {nullptr, nullptr, matrix_diag::Prepare, in Register_MATRIX_DIAG()
135 matrix_diag::Eval}; in Register_MATRIX_DIAG()
|
| D | Android.bp | 116 "matrix_diag.cc",
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 380 v_diag = array_ops.matrix_diag(v)
387 v_diag = array_ops.matrix_diag(v, k=offset)
395 v_diags = array_ops.matrix_diag(vecs[0], k=diags, align=align)
405 v_batch_diag = array_ops.matrix_diag(v_batch)
411 v_batch_diag = array_ops.matrix_diag(v_batch, k=offset)
424 v_diags = array_ops.matrix_diag(
488 v_diags = array_ops.matrix_diag(
508 v_diags = array_ops.matrix_diag(
527 v_diags = array_ops.matrix_diag(
541 array_ops.matrix_diag(0)
[all …]
|
| D | tridiagonal_matmul_op_test.py | 98 super_part = array_ops.pad(array_ops.matrix_diag(superdiag), super_pad)
99 main_part = array_ops.matrix_diag(maindiag)
100 sub_part = array_ops.pad(array_ops.matrix_diag(subdiag), sub_pad)
|
| D | self_adjoint_eig_op_test.py | 165 test_util.matmul_without_tf32(tf_v, array_ops.matrix_diag(tf_e)),
|
| D | eig_op_test.py | 175 math_ops.matmul(tf_v, array_ops.matrix_diag(tf_e)),
|
| D | svd_op_test.py | 119 diag_s = math_ops.cast(array_ops.matrix_diag(s), dtype=dtype_)
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixDiagV3.pbtxt | 7 name: "matrix_diag"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 746 mid = array_ops.matrix_diag(grad_e)
747 diag_grad_part = array_ops.matrix_diag(
764 vt, math_ops.matmul(array_ops.matrix_diag(grad_e), vt))
791 array_ops.matrix_diag(grad_e) +
799 array_ops.matrix_diag(grad_e),
825 grad_s_mat = array_ops.matrix_diag(grad_s)
870 s_mat = array_ops.matrix_diag(s)
886 s_inv_mat = array_ops.matrix_diag(_SafeReciprocal(s))
|
| D | linalg_ops_impl.py | 77 return array_ops.matrix_diag(diag_ones)
|
| D | array_grad.py | 390 return array_ops.matrix_diag(grad)
400 return array_ops.matrix_diag(
416 return array_ops.matrix_diag(
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_diag_test.py | 59 matrix = array_ops.matrix_diag(diag)
147 mat = array_ops.matrix_diag(diag_broadcast)
|
| D | linear_operator_lower_triangular_test.py | 52 tril = array_ops.matrix_diag(tril)
|
| D | linear_operator_low_rank_update_test.py | 127 base_diag_mat = array_ops.matrix_diag(base_diag)
130 diag_update_mat = array_ops.matrix_diag(diag_update)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_diag.py | 251 return array_ops.matrix_diag(self._diag)
|
| D | linalg_impl.py | 45 diag = array_ops.matrix_diag
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1038 below = array_ops.matrix_diag(_pad(below))[..., :-1, 1:]
1041 diag = array_ops.matrix_diag(diag)
1044 above = array_ops.matrix_diag(_pad(above))[..., 1:, :-1]
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 98 "kernels/matrix_diag.cc",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 99 from tensorflow.lite.testing.op_tests.matrix_diag import make_matrix_diag_tests
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 355 output = array_ops.matrix_diag(**params)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 383 return array_ops.matrix_diag(
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 856 array_ops.matrix_diag([1.0, 0.0, 1.0, 0.3333]), (1, 4, 4))
862 array_ops.matrix_diag([1.0, 1.0, 0.0, 0.3333]), (1, 4, 4))
|