| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 43 class ConvOp : public XlaOpKernel { class
45 explicit ConvOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvOp() function in tensorflow::__anon7c19f5bd0111::ConvOp
65 TF_DISALLOW_COPY_AND_ASSIGN(ConvOp);
68 class Conv2DOp : public ConvOp {
71 : ConvOp(ctx, /*num_spatial_dims=*/2, /*depthwise=*/false) {} in Conv2DOp()
76 class Conv3DOp : public ConvOp {
79 : ConvOp(ctx, /*num_spatial_dims=*/3, /*depthwise=*/false) {} in Conv3DOp()
84 class DepthwiseConv2DOp : public ConvOp {
87 : ConvOp(ctx, /*num_spatial_dims=*/2, /*depthwise=*/true) {} in DepthwiseConv2DOp()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Generalization.cpp | 93 : public LinalgGeneralizationPattern<GeneralizeConvOp, linalg::ConvOp> {
96 linalg::GenericOp createGenericOp(linalg::ConvOp, OpBuilder &rewriter) const;
149 linalg::GenericOp GeneralizeConvOp::createGenericOp(linalg::ConvOp convOp, in createGenericOp()
|
| D | Vectorization.cpp | 533 template <class ConvOp, int N>
534 LogicalResult ConvOpVectorization<ConvOp, N>::matchAndRewrite( in matchAndRewrite()
535 ConvOp op, PatternRewriter &rewriter) const { in matchAndRewrite()
603 template <typename ConvOp, unsigned N>
615 tilingPatterns.insert<LinalgTilingPattern<ConvOp>>( in populateVectorizationPatterns()
619 promotionPatterns.insert<LinalgPromotionPattern<ConvOp>>( in populateVectorizationPatterns()
629 vectorizationPatterns.insert<ConvOpVectorization<ConvOp, N>>(context, mask); in populateVectorizationPatterns()
|
| D | Loops.cpp | 319 Attribute getPadValueAttr<ConvOp>(Type type) { in getPadValueAttr()
327 static bool hasPadding(ConvOp convOp) { in hasPadding()
336 static void emitScalarImplementation(ArrayRef<Value> allIvs, ConvOp convOp) { in emitScalarImplementation()
358 Value padValue = std_constant(type, getPadValueAttr<ConvOp>(type)); in emitScalarImplementation()
528 .Case<CopyOp, FillOp, ConvOp, PoolingMaxOp, PoolingMinOp, in linalgOpToLoopsImpl()
|
| D | Fusion.cpp | 341 if (auto convOp = dyn_cast<linalg::ConvOp>(producer.getOperation())) { in isFusableInto()
346 if (auto convOp = dyn_cast<linalg::ConvOp>(consumer.getOperation())) { in isFusableInto()
|
| D | Promotion.cpp | 312 if (auto convOp = dyn_cast<linalg::ConvOp>(op.getOperation())) { in promoteSubViews()
|
| D | Tiling.cpp | 327 if (auto convOp = dyn_cast<linalg::ConvOp>(op.getOperation())) { in tileLinalgOpImpl()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 757 template <typename ConvOp, int N>
758 class ConvOpVectorization : public OpRewritePattern<ConvOp> {
759 using OpRewritePattern<ConvOp>::OpRewritePattern;
764 : OpRewritePattern<ConvOp>(context) { in ConvOpVectorization()
769 LogicalResult matchAndRewrite(ConvOp minOp,
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgOps.h | 35 class ConvOp; variable
|
| D | LinalgStructuredOps.td | 279 def ConvOp : PoolingBase_Op<"conv", [
281 // Despite having reductions, this manually defined ConvOp may only take
400 // Despite having reductions, this manually defined ConvOp may only take
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_hlo_to_lhlo_op.h | 50 MAP_HLO_TO_LHLO(ConvOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 60 class ConvertConvOp : public OpConversionPattern<mhlo::ConvOp> {
65 mhlo::ConvOp conv_op, ArrayRef<Value> args, in matchAndRewrite()
136 bool IsSamePadding(mhlo::ConvOp conv_op, int num_spatial_dims, in IsSamePadding()
158 void CreateConvOp(mhlo::ConvOp conv_op, ArrayRef<int64_t> strides, in CreateConvOp()
197 bool IsSupportedConvOp(mhlo::ConvOp conv_op) const { in IsSupportedConvOp()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgFusionTransforms.cpp | 42 LinalgTileAndFusePattern<ConvOp>>( in fillFusionPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 359 } else if (isa<mlir::mhlo::ConvOp, mlir::lmhlo::ConvOp>(op)) { in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 722 MakeAndReturn(ConvOp); in ImportInstructionImpl()
|
| D | mlir_hlo_to_hlo.cc | 712 LogicalResult ExportXlaOp(mlir::mhlo::ConvOp op, OpLoweringContext ctx) { in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 262 struct ConvToLinalgConverter : public OpConversionPattern<lmhlo::ConvOp> {
264 using OpConversionPattern<lmhlo::ConvOp>::OpConversionPattern;
269 lmhlo::ConvOp op, ArrayRef<Value> args, in matchAndRewrite()
354 rewriter.replaceOpWithNewOp<linalg::ConvOp>( in matchAndRewrite()
|
| D | hlo_legalize_to_lhlo.cc | 654 HloToLhloOpConverter<mhlo::ConvOp>, in populateHLOToLHLOConversionPattern()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/ |
| D | TosaOpBase.td | 75 // ConvOp and MatMulOp QuantizationAttrs follow a common design semantic where
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 1224 void ConvOp::getEffects( in getEffects()
1235 static LogicalResult verify(ConvOp op) { in verify()
1383 INSTANTIATE_WEIGHTED_POOLING_INPUT_INDEX(ConvOp)
1835 CANONICALIZERS_AND_FOLDERS(ConvOp)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 106 auto op = builder_.create<mlir::mhlo::ConvOp>( in ConvGeneralDilatedInternal()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/PowerPC/ |
| D | PPCISelDAGToDAG.cpp | 2575 SDValue ConvOp = WideRes; in tryEXTEND() local
2577 ConvOp = addExtOrTrunc(WideRes, Input32Bit ? ExtOrTruncConversion::Ext : in tryEXTEND()
2579 return ConvOp.getNode(); in tryEXTEND()
|
| D | PPCISelLowering.cpp | 7955 unsigned ConvOp = Signed ? (SinglePrec ? PPCISD::FCFIDS : PPCISD::FCFID) : in LowerINT_TO_FPDirectMove() local
7961 FP = DAG.getNode(ConvOp, dl, SinglePrec ? MVT::f32 : MVT::f64, FP); in LowerINT_TO_FPDirectMove()
7965 FP = DAG.getNode(ConvOp, dl, SinglePrec ? MVT::f32 : MVT::f64, FP); in LowerINT_TO_FPDirectMove()
13357 unsigned ConvOp = Signed ? in combineFPToIntToFP() local
13373 return DAG.getNode(ConvOp, dl, DstDouble ? MVT::f64 : MVT::f32, Ext); in combineFPToIntToFP()
13375 return DAG.getNode(ConvOp, dl, DstDouble ? MVT::f64 : MVT::f32, Ld); in combineFPToIntToFP()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1279 rewriter.replaceOpWithNewOp<ConvOp>(op, op.getType(), operands, in matchAndRewrite()
4237 Value result = rewriter.create<ConvOp>( in matchAndRewrite()
4442 Value result = rewriter.create<ConvOp>( in matchAndRewrite()
|
| /external/llvm-project/llvm/lib/Target/PowerPC/ |
| D | PPCISelDAGToDAG.cpp | 2535 SDValue ConvOp = WideRes; in tryEXTEND() local
2537 ConvOp = addExtOrTrunc(WideRes, Input32Bit ? ExtOrTruncConversion::Ext : in tryEXTEND()
2539 return ConvOp.getNode(); in tryEXTEND()
|