Searched refs:SetMatrixDiag (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 117 xla::XlaOp SetMatrixDiag(const xla::XlaOp input, const xla::XlaOp diag, in SetMatrixDiag() function
318 0, SetMatrixDiag(output, diag, output_shape, diag_rank, num_diags, in Compile()
525 0, SetMatrixDiag(input, diag, input_shape, diag_rank, num_diags, in Compile()
|