Searched refs:ExtractK (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ | ||
| D | legalize_tf.cc | 1432 LogicalResult ExtractK(TF::MatrixDiagPartV3Op op, int64_t (*k)[2]) const { in ExtractK() function in mlir::mhlo::__anon5ef947ee0111::ConvertMatrixDiagPartV3Op |