Searched refs:arg_bufs (Results 1 – 1 of 1) sorted by relevance
44 SmallVector<Value, 2> arg_bufs; in ApplySingleResultLhloCode() local46 arg_bufs.push_back(b->create<AllocOp>(loc, arg_type.cast<MemRefType>())); in ApplySingleResultLhloCode()49 b->create<StoreOp>(loc, operand.value(), arg_bufs[operand.index()]); in ApplySingleResultLhloCode()53 mapping.map(lhlo_block->getArguments(), arg_bufs); in ApplySingleResultLhloCode()58 return b->create<LoadOp>(loc, arg_bufs.back()); in ApplySingleResultLhloCode()