Searched refs:matrix_set_diag (Results 1 – 25 of 36) sorted by relevance
12
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 588 output = array_ops.matrix_set_diag(mat, v)
599 output = array_ops.matrix_set_diag(
610 output = array_ops.matrix_set_diag(mat, v)
617 output = array_ops.matrix_set_diag(mat, v)
628 output = array_ops.matrix_set_diag(
645 output = array_ops.matrix_set_diag(mat_batch, v_batch)
656 output = array_ops.matrix_set_diag(
678 output = array_ops.matrix_set_diag(mat_batch, v_batch)
690 output = array_ops.matrix_set_diag(
698 array_ops.matrix_set_diag(0, [0])
[all …]
|
| D | cholesky_op_test.py | 45 middle = array_ops.matrix_set_diag(middle,
59 middle = array_ops.matrix_set_diag(middle,
|
| D | lu_op_test.py | 89 lower = array_ops.matrix_set_diag(lower, ones_diag)
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | matrix_set_diag.cc | 27 namespace matrix_set_diag { namespace
147 static TfLiteRegistration r = {nullptr, nullptr, matrix_set_diag::Prepare, in Register_MATRIX_SET_DIAG()
148 matrix_set_diag::Eval}; in Register_MATRIX_SET_DIAG()
|
| D | Android.bp | 117 "matrix_set_diag.cc",
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixSetDiagV3.pbtxt | 95 tf.matrix_set_diag(input, diagonal)
104 tf.matrix_set_diag(input, diagonal, k = 1)
121 tf.matrix_set_diag(input, diagonals, k = (-1, 2))
138 tf.matrix_set_diag(input, diagonals, k = (-1, 2), align="LEFT_RIGHT")
|
| D | api_def_MatrixSetDiagV2.pbtxt | 72 tf.matrix_set_diag(diagonal) ==> [[[1, 7, 7, 7], # Output shape: (2, 3, 4)
80 tf.matrix_set_diag(diagonal, k = 1)
93 tf.matrix_set_diag(diagonals, k = (-1, 0))
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 429 array_ops.matrix_set_diag(matrix_i, diag_i),
430 array_ops.matrix_set_diag(matrices[0, ...], diag_i),
431 array_ops.matrix_set_diag(matrix_i, diags[0, ...]),
438 array_ops.matrix_set_diag(matrix_i, band_i, k=k, align=align),
439 array_ops.matrix_set_diag(
441 array_ops.matrix_set_diag(
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixSetDiagV3.pbtxt | 7 name: "matrix_set_diag"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops_impl.py | 80 return array_ops.matrix_set_diag(zero_matrix, diag_ones)
|
| D | array_grad.py | 392 return array_ops.matrix_set_diag(array_ops.zeros_like(op.inputs[0]), grad)
406 return array_ops.matrix_set_diag(
423 return array_ops.matrix_set_diag(
445 grad_input = array_ops.matrix_set_diag(
481 grad_input = array_ops.matrix_set_diag(
518 grad_input = array_ops.matrix_set_diag(
|
| D | linalg_grad.py | 476 middle = array_ops.matrix_set_diag(middle,
740 f = array_ops.matrix_set_diag(
784 f = array_ops.matrix_set_diag(
805 grad_a = array_ops.matrix_set_diag(grad_a,
882 f = array_ops.matrix_set_diag(
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_householder_test.py | 73 matrix = array_ops.matrix_set_diag(
|
| D | linear_operator_full_matrix_test.py | 81 tril = array_ops.matrix_set_diag(tril, diag)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_diag.py | 259 return array_ops.matrix_set_diag(x, new_diag)
|
| D | linear_operator_identity.py | 412 return array_ops.matrix_set_diag(mat, new_diag)
754 return array_ops.matrix_set_diag(mat, new_diag)
|
| D | linear_operator_householder.py | 247 return array_ops.matrix_set_diag(
|
| D | linalg_impl.py | 58 set_diag = array_ops.matrix_set_diag
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | multinomial.py | 297 return array_ops.matrix_set_diag(
|
| D | dirichlet_multinomial.py | 317 return array_ops.matrix_set_diag(
|
| D | dirichlet.py | 272 return array_ops.matrix_set_diag(
|
| D | util.py | 579 transformed_mat = array_ops.matrix_set_diag(matrix, transformed_diag)
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 99 "kernels/matrix_set_diag.cc",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 100 from tensorflow.lite.testing.op_tests.matrix_set_diag import make_matrix_set_diag_tests
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 544 output = array_ops.matrix_set_diag(**params)
|
12