Searched refs:lhlo_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 234 LhloOp lhlo_op, ArrayRef<Value> args, in matchAndRewrite() argument
236 auto loc = lhlo_op.getLoc(); in matchAndRewrite()
238 lhlo_op.getOperand(0).getType().template dyn_cast<ShapedType>(); in matchAndRewrite()
245 auto lhs = rewriter.create<LoadOp>(loc, lhlo_op.lhs()); in matchAndRewrite()
246 auto rhs = rewriter.create<LoadOp>(loc, lhlo_op.rhs()); in matchAndRewrite()
249 lhlo_op, arg_type.getElementType(), llvm::ArrayRef<Value>{lhs, rhs}, in matchAndRewrite()
251 rewriter.create<StoreOp>(loc, op_result, lhlo_op.out()); in matchAndRewrite()
252 rewriter.eraseOp(lhlo_op); in matchAndRewrite()
|