Searched refs:matrix_set_diag (Results 1 – 25 of 36) sorted by relevance
12
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | diag_op_test.py | 584 output = array_ops.matrix_set_diag(mat, v)
595 output = array_ops.matrix_set_diag(
606 output = array_ops.matrix_set_diag(mat, v)
613 output = array_ops.matrix_set_diag(mat, v)
624 output = array_ops.matrix_set_diag(
641 output = array_ops.matrix_set_diag(mat_batch, v_batch)
652 output = array_ops.matrix_set_diag(
674 output = array_ops.matrix_set_diag(mat_batch, v_batch)
686 output = array_ops.matrix_set_diag(
694 array_ops.matrix_set_diag(0, [0])
[all …]
|
| /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 | 121 "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 | 454 array_ops.matrix_set_diag(matrix_i, diag_i),
455 array_ops.matrix_set_diag(matrices[0, ...], diag_i),
456 array_ops.matrix_set_diag(matrix_i, diags[0, ...]),
463 array_ops.matrix_set_diag(matrix_i, band_i, k=k, align=align),
464 array_ops.matrix_set_diag(
466 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 | 77 return array_ops.matrix_set_diag(zero_matrix, diag_ones)
|
| D | linalg_grad.py | 476 middle = array_ops.matrix_set_diag(middle,
744 f = array_ops.matrix_set_diag(
788 f = array_ops.matrix_set_diag(
809 grad_a = array_ops.matrix_set_diag(grad_a,
887 f = array_ops.matrix_set_diag(
|
| D | array_grad.py | 402 return array_ops.matrix_set_diag(array_ops.zeros_like(op.inputs[0]), grad)
416 return array_ops.matrix_set_diag(
433 return array_ops.matrix_set_diag(
455 grad_input = array_ops.matrix_set_diag(
491 grad_input = array_ops.matrix_set_diag(
528 grad_input = array_ops.matrix_set_diag(
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_householder_test.py | 77 matrix = array_ops.matrix_set_diag(
|
| D | cholesky_op_test.py | 40 middle = array_ops.matrix_set_diag(middle,
54 middle = array_ops.matrix_set_diag(middle,
|
| D | linear_operator_full_matrix_test.py | 78 tril = array_ops.matrix_set_diag(tril, diag)
|
| D | lu_op_test.py | 86 lower = array_ops.matrix_set_diag(lower, ones_diag)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_diag.py | 254 return array_ops.matrix_set_diag(x, new_diag)
|
| D | linear_operator_householder.py | 241 return array_ops.matrix_set_diag(
|
| D | linear_operator_identity.py | 409 return array_ops.matrix_set_diag(mat, new_diag)
768 return array_ops.matrix_set_diag(mat, new_diag)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 562 output = array_ops.matrix_set_diag(**params)
565 output = gen_array_ops.matrix_set_diag(**params)
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | multinomial.py | 294 return array_ops.matrix_set_diag(
|
| D | dirichlet_multinomial.py | 314 return array_ops.matrix_set_diag(
|
| D | dirichlet.py | 269 return array_ops.matrix_set_diag(
|
| D | util.py | 575 transformed_mat = array_ops.matrix_set_diag(matrix, transformed_diag)
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 119 "kernels/matrix_set_diag.cc",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 112 from tensorflow.lite.testing.op_tests.matrix_set_diag import make_matrix_set_diag_tests
|
| D | build_def.bzl | 94 "matrix_set_diag",
|
12