Home
last modified time | relevance | path

Searched refs:tfl_mm_op (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tfl.cc1355 auto tfl_mm_op = cast<TFL::BatchMatMulOp>(op); in matchAndRewrite() local
1356 auto result_ty = tfl_mm_op.getType().cast<ShapedType>(); in matchAndRewrite()
1357 Value lhs = tfl_mm_op.x(); in matchAndRewrite()
1358 Value rhs = tfl_mm_op.y(); in matchAndRewrite()
1361 bool transpose_lhs = tfl_mm_op.adj_x(); in matchAndRewrite()
1362 bool transpose_rhs = tfl_mm_op.adj_y(); in matchAndRewrite()