| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 55 typedef std::pair<TF::Conv2DOp, int64_t> Conv2DWithBlockSize;
195 void HandleConv2DStride(TF::Conv2DOp conv2d) { in HandleConv2DStride()
207 void HandleConv2DInput(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DInput()
262 void HandleConv2DFilter(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DFilter()
572 Optional<BlockArgumentInfo> GetConv2DInputArgNum(TF::Conv2DOp conv2d) { in GetConv2DInputArgNum()
582 void HandleFirstConvolution(TF::Conv2DOp conv2d, int64_t block_size) { in HandleFirstConvolution()
634 int32_t GetConv2DBlockSize(TF::Conv2DOp conv2d) { in GetConv2DBlockSize()
666 TF::Conv2DOp first_conv; in runOnOperation()
674 auto conv2d_result = device_func.walk([&](TF::Conv2DOp conv2d) { in runOnOperation()
|
| D | fused_kernel_matcher.cc | 186 : public FuseContractionWithBiasAdd<Conv2DOp, _FusedConv2DOp> {
188 using FuseContractionWithBiasAdd<Conv2DOp,
193 bool AreFuseCompatible(Conv2DOp conv, BiasAddOp bias_add, in AreFuseCompatible()
|
| D | optimize.td | 45 def DefinedByConv2D : Constraint<CPred<"llvm::isa_and_nonnull<mlir::TF::Conv2DOp>($0.getDefiningOp(…
|
| D | legalize_hlo.cc | 186 rewriter.replaceOpWithNewOp<Conv2DOp>( in CreateConvOp()
|
| /external/llvm-project/mlir/test/lib/Dialect/Tosa/ |
| D | TosaTestPasses.cpp | 103 : RewritePattern(tosa::Conv2DOp::getOperationName(), 1, context) {} in ConvertTosaConv2DOp()
112 auto tosaConv2DOp = cast<tosa::Conv2DOp>(op); in matchAndRewrite()
148 auto newTosaConv2DOp = rewriter.create<tosa::Conv2DOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 68 class Conv2DOp : public ConvOp { class
70 explicit Conv2DOp(OpKernelConstruction* ctx) in Conv2DOp() function in tensorflow::__anon3d40b8640111::Conv2DOp
74 Conv2DOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | fuse_bias_tf.cc | 76 TF::Conv2DOp tf_conv2d_op = in matchAndRewrite()
77 dyn_cast_or_null<TF::Conv2DOp>(value.getDefiningOp()); in matchAndRewrite()
|
| D | legalize_tfl.cc | 1100 auto tfl_conv2d_op = cast<TFL::Conv2DOp>(op); in matchAndRewrite()
1157 auto a1_conv2d_op = rewriter.create<tosa::Conv2DOp>( in matchAndRewrite()
|
| D | legalize_tf.cc | 816 auto tf_conv2d_op = cast<TF::Conv2DOp>(op); in matchAndRewrite()
|
| D | legalize_common.cc | 2859 .create<tosa::Conv2DOp>(op->getLoc(), output_type, input, in convertTFConv2DCommon()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.cc | 32 patterns.insert<ConvertTFDilatedConvOp<TF::Conv2DOp>, in runOnFunction()
|
| D | optimize.cc | 576 if (isa<TFL::Conv2DOp>(mul_op_lhs)) { in matchAndRewrite()
609 using FuseConv2DAndMulWithQDQs = FuseAffinOpAndMulWithQDQs<TFL::Conv2DOp>;
841 if (!isa_and_nonnull<TFL::Conv2DOp, TFL::DepthwiseConv2DOp, in CanFuseAffineOp()
952 using FuseBinaryOpToFollowingConv2D = FuseBinaryOpToFollowingAffineOp<Conv2DOp>;
|
| D | prepare_tf.cc | 383 class ConvertTFConv2D : public ConvertTFConvOp<ConvertTFConv2D, TF::Conv2DOp> {
385 using BaseType = ConvertTFConvOp<ConvertTFConv2D, TF::Conv2DOp>;
394 TFL::Conv2DOp createTFLOp(ConvertTFConvOpMatchState *state, in createTFLOp()
399 return rewriter.create<TFL::Conv2DOp>( in createTFLOp()
1464 patterns.insert<ConvertTFDilatedConvOp<TF::Conv2DOp>, FusedBatchNormV3Pat, in runOnFunction()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 509 class Conv2DOp : public BinaryOp<T> { class
511 explicit Conv2DOp(OpKernelConstruction* context) : BinaryOp<T>(context) { in Conv2DOp() function in tensorflow::Conv2DOp
596 TF_DISALLOW_COPY_AND_ASSIGN(Conv2DOp);
602 Conv2DOp<CPUDevice, T>);
1187 Conv2DOp<GPUDevice, Eigen::half>);
1190 Conv2DOp<GPUDevice, float>);
1193 Conv2DOp<GPUDevice, double>);
1196 Conv2DOp<GPUDevice, int32>);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1514 OpT, Conv2DOp, Conv3DOp>::value>::type * = nullptr>
1516 int num_spatial_dims = std::is_same<OpT, Conv2DOp>() ? 2 : 3; in Verify()
1547 LogicalResult Conv2DOp::UpdateDataFormat(StringRef data_format) { in UpdateDataFormat()
1678 LogicalResult Conv2DOp::inferReturnTypes( in inferReturnTypes()
1696 StringRef Conv2DOp::GetOptimalLayout(const RuntimeDevices &devices) { in GetOptimalLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 944 void Conv2DOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1285 using ConvertConv2DOp = ConvertConvOp<TF::Conv2DOp, /*num_spatial_dims=*/2>;
|