Searched refs:set_diag (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_MatrixSetDiagV3.pbtxt | 4 name: "linalg.set_diag"
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | matrix_set_diag.py | 46 outs = tf.linalg.set_diag(input_tensor, diag_tensor)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 216 name: "set_diag"
217 …, \'name\', \'k\', \'align\'], varargs=None, keywords=None, defaults=[\'set_diag\', \'0\', \'RIGHT…
|
| D | tensorflow.pbtxt | 1661 …, \'name\', \'k\', \'align\'], varargs=None, keywords=None, defaults=[\'set_diag\', \'0\', \'RIGHT…
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 58 set_diag = array_ops.matrix_set_diag variable
1003 lower = set_diag(
1126 lower = set_diag(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 526 eyem = _linalg.set_diag(array_ops.zeros_like(m), _linalg.diag_part(m))
|