Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc385 std::array<Value, 3> bufferArgs = {operands[0], operands[1], {}}; in matchAndRewrite() local
388 bufferArgs[2] = InsertAlloc(op->getLoc(), result, &rewriter); in matchAndRewrite()
395 bufferArgs[2] = InsertDynamicAllocAndDealloc( in matchAndRewrite()
399 rewriter.create<lmhlo::DotOp>(op->getLoc(), llvm::None, bufferArgs, in matchAndRewrite()
401 rewriter.replaceOp(op, bufferArgs[2]); in matchAndRewrite()