| /external/llvm-project/mlir/include/mlir/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.h | 44 class CopyOpToLibraryCallRewrite : public OpRewritePattern<CopyOp> {
46 using OpRewritePattern<CopyOp>::OpRewritePattern;
47 LogicalResult matchAndRewrite(CopyOp op,
54 class CopyTransposeRewrite : public OpRewritePattern<CopyOp> {
56 using OpRewritePattern<CopyOp>::OpRewritePattern;
57 LogicalResult matchAndRewrite(CopyOp op,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | debug_ops.cc | 27 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_CPU), CopyOp);
29 REGISTER_KERNEL_BUILDER(Name("CopyHost").Device(DEVICE_CPU), CopyOp);
32 REGISTER_KERNEL_BUILDER(Name("Copy").Device(DEVICE_GPU), CopyOp);
38 CopyOp);
|
| D | inplace_ops.cc | 339 class CopyOp : public CopyOpBase { class
341 explicit CopyOp(OpKernelConstruction* ctx) : CopyOpBase(ctx) {} in CopyOp() function in tensorflow::__anon4b908b380211::CopyOp
416 REGISTER_KERNEL_BUILDER(Name("DeepCopy").Device(DEVICE_CPU), CopyOp<CPUDevice>);
450 CopyOp<GPUDevice>);
490 CopyOp<CPUDevice>);
|
| D | debug_ops.h | 47 class CopyOp : public OpKernel {
49 explicit CopyOp(OpKernelConstruction* context) : OpKernel(context) { in CopyOp() function
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 105 if (!isa<LinalgOp>(op) || isa<CopyOp>(op) || isa<IndexedGenericOp>(op)) in matchAndRewrite()
120 CopyOp op, PatternRewriter &rewriter) const { in matchAndRewrite()
140 CopyOp op, PatternRewriter &rewriter) const { in matchAndRewrite()
157 rewriter.replaceOpWithNewOp<CopyOp>(op, in, out); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Vectorization.cpp | 286 if (isa<linalg::FillOp, linalg::CopyOp>(op)) in vectorizeLinalgOpPrecondition()
308 if (auto copyOp = dyn_cast<linalg::CopyOp>(op)) { in vectorizeLinalgOp()
430 CopyOp copyOp; in matchAndRewrite()
432 if (auto newCopyOp = dyn_cast<CopyOp>(u.getOwner())) { in matchAndRewrite()
501 CopyOp copyOp; in matchAndRewrite()
503 if (auto newCopyOp = dyn_cast<CopyOp>(u.getOwner())) { in matchAndRewrite()
|
| D | Bufferize.cpp | 39 b.create<linalg::CopyOp>(loc, memref, alloc); in cloneMemref()
269 rewriter.create<linalg::CopyOp>(op.getLoc(), subView, alloc); in matchAndRewrite()
312 rewriter.create<linalg::CopyOp>(op.getLoc(), sourceMemRef, subview); in matchAndRewrite()
|
| D | Loops.cpp | 184 static void emitScalarImplementation(ArrayRef<Value> allIvs, CopyOp copyOp) { in emitScalarImplementation()
528 .Case<CopyOp, FillOp, ConvOp, PoolingMaxOp, PoolingMinOp, in linalgOpToLoopsImpl()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/EDSC/ |
| D | Intrinsics.h | 18 using linalg_copy = OperationBuilder<linalg::CopyOp>;
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgTransforms.cpp | 172 LinalgVectorizationPattern<CopyOp>, in applyPatterns()
240 LinalgVectorizationPattern<CopyOp>>(ctx); in fillL1TilingAndMatmulToVectorPatterns()
277 b.create<CopyOp>(src.getLoc(), src, dst); in copyCallBackFn()
474 LinalgVectorizationPattern<FillOp>, LinalgVectorizationPattern<CopyOp>, in applyLinalgToVectorPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_hlo_to_lhlo_op.h | 52 MAP_HLO_TO_LHLO(CopyOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 265 rewriter.create<lmhlo::CopyOp>(loc, result, result_buffer); in matchAndRewrite()
476 rewriter.create<lmhlo::CopyOp>(loc, operand, in matchAndRewrite()
494 rewriter.replaceOpWithNewOp<lmhlo::CopyOp>(op, llvm::None, operands.front(), in matchAndRewrite()
656 HloToLhloOpConverter<mhlo::CopyOp>, in populateHLOToLHLOConversionPattern()
|
| D | legalize_to_linalg.cc | 808 rewriter.replaceOpWithNewOp<linalg::CopyOp>(reshape_op, reshape_buffer, in matchAndRewrite()
823 rewriter.replaceOpWithNewOp<linalg::CopyOp>(reshape_op, reshape_buffer, in matchAndRewrite()
1065 rewriter.create<linalg::CopyOp>(loc, linalg_slice, slice_op.getOperand(1)); in matchAndRewrite()
1397 PointwiseToLinalgConverter<lmhlo::CopyOp>, in populateLHLOToLinalgConversionPattern()
1524 PointwiseToLinalgConverter<mhlo::CopyOp, false>, in populateHLOToLinalgConversionPattern()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | BufferResultsToOutParams.cpp | 73 builder.create<linalg::CopyOp>(op.getLoc(), std::get<0>(t), in updateReturnOps()
|
| D | BufferDeallocation.cpp | 412 builder.create<linalg::CopyOp>(terminator->getLoc(), sourceValue, alloc); in introduceBufferCopy()
|
| /external/llvm-project/mlir/test/Transforms/ |
| D | buffer-deallocation.mlir | 196 // exit block after CopyOp since %1 is an alias for %0 and %arg1. Furthermore,
232 // exit block after CopyOp since %1 is an alias for %0 and %arg1.
260 // %7 should happen after CopyOp.
522 // inserted after CopyOp.
569 // deallocating. It should dealloc %y after CopyOp.
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 440 } else if (isa<mlir::mhlo::CopyOp, mlir::lmhlo::CopyOp>(op)) { in MhloToHloOpcode()
|
| /external/llvm/lib/Transforms/Instrumentation/ |
| D | MemorySanitizer.cpp | 2077 Value *CopyOp, *ConvertOp; in handleVectorConvertIntrinsic() local
2083 CopyOp = I.getArgOperand(0); in handleVectorConvertIntrinsic()
2088 CopyOp = nullptr; in handleVectorConvertIntrinsic()
2119 if (CopyOp) { in handleVectorConvertIntrinsic()
2120 assert(CopyOp->getType() == I.getType()); in handleVectorConvertIntrinsic()
2121 assert(CopyOp->getType()->isVectorTy()); in handleVectorConvertIntrinsic()
2122 Value *ResultShadow = getShadow(CopyOp); in handleVectorConvertIntrinsic()
2130 setOrigin(&I, getOrigin(CopyOp)); in handleVectorConvertIntrinsic()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Instrumentation/ |
| D | MemorySanitizer.cpp | 2615 Value *CopyOp, *ConvertOp; in handleVectorConvertIntrinsic() local
2622 CopyOp = I.getArgOperand(0); in handleVectorConvertIntrinsic()
2627 CopyOp = nullptr; in handleVectorConvertIntrinsic()
2658 if (CopyOp) { in handleVectorConvertIntrinsic()
2659 assert(CopyOp->getType() == I.getType()); in handleVectorConvertIntrinsic()
2660 assert(CopyOp->getType()->isVectorTy()); in handleVectorConvertIntrinsic()
2661 Value *ResultShadow = getShadow(CopyOp); in handleVectorConvertIntrinsic()
2669 setOrigin(&I, getOrigin(CopyOp)); in handleVectorConvertIntrinsic()
|
| /external/llvm-project/llvm/lib/Transforms/Instrumentation/ |
| D | MemorySanitizer.cpp | 2740 Value *CopyOp, *ConvertOp; in handleVectorConvertIntrinsic() local
2748 CopyOp = I.getArgOperand(0); in handleVectorConvertIntrinsic()
2753 CopyOp = nullptr; in handleVectorConvertIntrinsic()
2784 if (CopyOp) { in handleVectorConvertIntrinsic()
2785 assert(CopyOp->getType() == I.getType()); in handleVectorConvertIntrinsic()
2786 assert(CopyOp->getType()->isVectorTy()); in handleVectorConvertIntrinsic()
2787 Value *ResultShadow = getShadow(CopyOp); in handleVectorConvertIntrinsic()
2795 setOrigin(&I, getOrigin(CopyOp)); in handleVectorConvertIntrinsic()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 1169 void CopyOp::getEffects( in getEffects()
1178 static LogicalResult verify(CopyOp op) { in verify()
1839 CANONICALIZERS_AND_FOLDERS(CopyOp)
|
| /external/llvm-project/llvm/lib/CodeGen/ |
| D | MachineInstr.cpp | 215 MachineOperand CopyOp(Op); in addOperand() local
216 return addOperand(MF, CopyOp); in addOperand()
|
| /external/llvm/lib/CodeGen/ |
| D | MachineInstr.cpp | 776 MachineOperand CopyOp(Op); in addOperand() local
777 return addOperand(MF, CopyOp); in addOperand()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/CodeGen/ |
| D | MachineInstr.cpp | 207 MachineOperand CopyOp(Op); in addOperand() local
208 return addOperand(MF, CopyOp); in addOperand()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 657 if (auto copy = mlir::dyn_cast<mlir::lmhlo::CopyOp>(input.op)) { in EmitUsingElementalIrEmitter()
661 auto fused_copy = b.create<mlir::mhlo::CopyOp>(loc, operand); in EmitUsingElementalIrEmitter()
2088 auto copy = mlir::cast<mlir::lmhlo::CopyOp>(input.op); in EmitCopyForMlir()
4883 if (auto copy = mlir::dyn_cast<mlir::lmhlo::CopyOp>(op)) { in EmitHlo021Tile()
5090 CHECK((mlir::isa<mlir::lmhlo::FusionOp, mlir::lmhlo::CopyOp>(input.op))); in CheckAndEmitHloWithTile021()
|