Home
last modified time | relevance | path

Searched defs:matchAndRewrite (Results 1 – 25 of 155) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc143 LogicalResult ConvertTFReluOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFReluOp
161 LogicalResult ConvertTFRelu6Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFRelu6Op
177 LogicalResult ConvertTFEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFEqualOp
191 LogicalResult ConvertTFNotEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFNotEqualOp
212 LogicalResult ConvertTFGreaterOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFGreaterOp
226 LogicalResult ConvertTFGreaterEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFGreaterEqualOp
241 LogicalResult ConvertTFAddOp::matchAndRewrite(Operation* op, in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFAddOp
255 LogicalResult ConvertTFAddV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFAddV2Op
270 LogicalResult ConvertTFAddNOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFAddNOp
295 LogicalResult ConvertTFSubOp::matchAndRewrite(Operation* op, in matchAndRewrite() function in mlir::tosa::__anon4375bd360111::ConvertTFSubOp
[all …]
Dlegalize_tfl.cc177 LogicalResult ConvertTFLReluOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLReluOp
227 LogicalResult ConvertTFLRelu1Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLRelu1Op
282 LogicalResult ConvertTFLRelu6Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLRelu6Op
390 LogicalResult ConvertTFLEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLEqualOp
403 LogicalResult ConvertTFLNotEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLNotEqualOp
420 LogicalResult ConvertTFLGreaterOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLGreaterOp
433 LogicalResult ConvertTFLGreaterEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLGreaterEqualOp
446 LogicalResult ConvertTFLLessOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLLessOp
458 LogicalResult ConvertTFLLessEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLLessEqualOp
575 LogicalResult ConvertTFLAddOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anonb24e82620111::ConvertTFLAddOp
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_tf_framework.cc36 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon099c28360111::FuncOpConverter
76 LogicalResult matchAndRewrite( in matchAndRewrite() function
108 LogicalResult matchAndRewrite( in matchAndRewrite() function
131 LogicalResult matchAndRewrite( in matchAndRewrite() function
147 LogicalResult matchAndRewrite( in matchAndRewrite() function
164 LogicalResult matchAndRewrite( in matchAndRewrite() function
Dtf_framework_legalize_to_llvm.cc112 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::TFAllocOpConverter
226 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::TFDeallocOpConverter
259 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::JITCompileFromStrOpConverter
322 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::JITExecuteOpConverter
405 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::ReportErrorOpConverter
460 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::NullContextOpConverter
472 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::NullMemRefOpConverter
551 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anon4f566a220111::IsValidMemRefOpConverter
Dshape_simplification.cc45 LogicalResult matchAndRewrite(BroadcastOp op, in matchAndRewrite() function
144 LogicalResult matchAndRewrite(tensor::ExtractOp op, in matchAndRewrite() function
171 LogicalResult matchAndRewrite(tensor::ExtractOp op, in matchAndRewrite() function
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
Dbufferize_tiled_loop.cc119 LogicalResult matchAndRewrite( in matchAndRewrite() function
135 LogicalResult matchAndRewrite( in matchAndRewrite() function
180 LogicalResult matchAndRewrite( in matchAndRewrite() function
239 LogicalResult matchAndRewrite( in matchAndRewrite() function
263 LogicalResult matchAndRewrite( in matchAndRewrite() function
283 LogicalResult matchAndRewrite( in matchAndRewrite() function
333 LogicalResult matchAndRewrite( in matchAndRewrite() function
350 LogicalResult matchAndRewrite( in matchAndRewrite() function
Dfusion.cc42 LogicalResult matchAndRewrite(tensor::ExtractOp extract, in matchAndRewrite() function
58 LogicalResult matchAndRewrite(tensor::DimOp op, in matchAndRewrite() function
126 LogicalResult matchAndRewrite(MaterializeOp op, in matchAndRewrite() function
Dgml_st_to_scf.cc38 LogicalResult matchAndRewrite(LoopOp loop, in matchAndRewrite() function
100 LogicalResult matchAndRewrite(ParallelOp loop, in matchAndRewrite() function
125 LogicalResult matchAndRewrite(ForOp loop, in matchAndRewrite() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtfg-to-tfe.cc139 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertGraphOp
174 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertGraphFuncOp
243 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertReturnOp
257 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertControlTriggerOp
277 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertEnterOp
297 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertExitOp
317 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertLoopCondOp
337 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertMergeOp
357 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertSwitchOp
377 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon249176490111::ConvertSwitchNOp
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc204 LogicalResult ConvertTFConcatV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFConcatV2Op
226 LogicalResult ConvertTFMatMulOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFMatMulOp
274 LogicalResult ConvertTFPackOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFPackOp
289 LogicalResult ConvertTFSplitOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFSplitOp
302 LogicalResult ConvertTFSplitVOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFSplitVOp
315 LogicalResult ConvertTFUnpackOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFUnpackOp
329 LogicalResult ConvertTFConv3DOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFConv3DOp
369 LogicalResult ConvertTFConv3DBackpropInputV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFConv3DBackpropInputV2Op
476 LogicalResult ConvertTFMatrixDiagV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFMatrixDiagV2Op
483 LogicalResult ConvertTFMatrixDiagV3Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon4283b78d0111::ConvertTFMatrixDiagV3Op
[all …]
Dlegalize_hashtables.cc53 LogicalResult matchAndRewrite(TF::HashTableV2Op hashtable_op, in matchAndRewrite() function in mlir::TFL::__anon48227b320111::LegalizeHashTableOpPattern
77 LogicalResult matchAndRewrite(TF::LookupTableFindV2Op find_op, in matchAndRewrite() function in mlir::TFL::__anon48227b320111::LegalizeHashTableFindOpPattern
95 LogicalResult matchAndRewrite(TF::LookupTableImportV2Op import_op, in matchAndRewrite() function in mlir::TFL::__anon48227b320111::LegalizeHashTableImportOpPattern
113 LogicalResult matchAndRewrite(TF::LookupTableSizeV2Op size_op, in matchAndRewrite() function in mlir::TFL::__anon48227b320111::LegalizeHashTableSizeOpPattern
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dmerge_assuming_ops.cc52 LogicalResult matchAndRewrite(shape::ShapeOfOp op, in matchAndRewrite() function
80 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function
182 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function
194 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
242 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
264 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function
336 LogicalResult matchAndRewrite(shape::AssumingOp op, in matchAndRewrite() function
403 LogicalResult matchAndRewrite(shape::CstrBroadcastableOp op, in matchAndRewrite() function
Dtest_infer_shaped_type_pass.cc35 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
67 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
105 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
Dlegalize_shape_computations.cc76 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function in mlir::__anon933c4f4c0111::MhloElementwiseConverter
114 LogicalResult matchAndRewrite(mhlo::ConcatenateOp op, in matchAndRewrite() function in mlir::__anon933c4f4c0111::ConcatenateConverter
147 LogicalResult matchAndRewrite(mhlo::GetDimensionSizeOp op, in matchAndRewrite() function in mlir::__anon933c4f4c0111::GetDimSizeConverter
169 LogicalResult matchAndRewrite(mhlo::ReshapeOp op, in matchAndRewrite() function in mlir::__anon933c4f4c0111::ReshapeConverter
Dhlo_legalize_to_lhlo.cc144 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anona600afe90111::HloToLhloOpConverter
167 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anona600afe90111::HloToLhloOpConverter
193 LogicalResult matchAndRewrite( in matchAndRewrite() function
219 LogicalResult matchAndRewrite( in matchAndRewrite() function
260 LogicalResult matchAndRewrite( in matchAndRewrite() function
324 LogicalResult matchAndRewrite( in matchAndRewrite() function
Dlegalize_control_flow.cc81 LogicalResult matchAndRewrite( in matchAndRewrite() function
112 LogicalResult matchAndRewrite( in matchAndRewrite() function
169 LogicalResult matchAndRewrite( in matchAndRewrite() function
Dlegalize_mhlo_to_thlo.cc50 LogicalResult matchAndRewrite(mhlo::ConcatenateOp op, in matchAndRewrite() function
119 LogicalResult matchAndRewrite(mhlo::DynamicBroadcastInDimOp op, in matchAndRewrite() function
186 LogicalResult matchAndRewrite(mhlo::GatherOp op, in matchAndRewrite() function
233 LogicalResult matchAndRewrite(mhlo::ScatterOp op, in matchAndRewrite() function
Dlegalize_to_standard.cc42 LogicalResult matchAndRewrite(mhlo::CompareOp op, in matchAndRewrite() function in mlir::mhlo::__anona244da490211::CompareIConvert
90 LogicalResult matchAndRewrite(mhlo::CompareOp op, in matchAndRewrite() function in mlir::mhlo::__anona244da490211::CompareFConvert
142 LogicalResult matchAndRewrite(mhlo::IotaOp op, in matchAndRewrite() function in mlir::mhlo::__anona244da490211::ConvertIotaOp
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc691 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::EvaluateConstant
792 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeShapeOp
831 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeSizeOp
866 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeRankOp
902 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeTensorArraySizeV3Op
943 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeShapeNOp
989 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeBroadcastGradientArgsOp
1104 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeReductionIndices
1181 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeFillNode
1222 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::MaterializeConstantValuedNode
[all …]
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
DTestUtils.cpp46 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
79 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
117 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc206 LogicalResult LowerPackIntoConcatReshape::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::tac::LowerPackIntoConcatReshape
263 LogicalResult SquaredDifference::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::tac::SquaredDifference
287 LogicalResult UnrollSplit::matchAndRewrite(TFL::SplitOp split_op, in matchAndRewrite() function in mlir::TFL::tac::UnrollSplit
317 LogicalResult UnrollSplitV::matchAndRewrite(TFL::SplitVOp splitv_op, in matchAndRewrite() function in mlir::TFL::tac::UnrollSplitV
359 LogicalResult EnsureBiasForConv2d::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::tac::EnsureBiasForConv2d
367 LogicalResult PadSlice::matchAndRewrite(TFL::SliceOp slice_op, in matchAndRewrite() function in mlir::TFL::tac::PadSlice
469 LogicalResult FullyConnectedToConv::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::tac::FullyConnectedToConv
529 LogicalResult PadConcat::matchAndRewrite(TFL::ConcatenationOp concat_op, in matchAndRewrite() function in mlir::TFL::tac::PadConcat
601 LogicalResult ReduceMeanToAvgPool::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::tac::ReduceMeanToAvgPool
666 LogicalResult InsertRequantForReduceMean::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::tac::InsertRequantForReduceMean
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
Dlegalize_to_tensor_op.cc47 LogicalResult matchAndRewrite(ExtractOp extract, in matchAndRewrite() function
66 LogicalResult matchAndRewrite(ShapeOfOp shapeOf, in matchAndRewrite() function
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tosa/lib/Transforms/
Dlegalize_mhlo.cc48 LogicalResult matchAndRewrite(mhlo::CompareOp op, in matchAndRewrite() function
78 LogicalResult matchAndRewrite(mhlo::DotOp op, in matchAndRewrite() function
169 LogicalResult matchAndRewrite(mhlo::ReduceOp op, in matchAndRewrite() function
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dlmhlo_to_jitrt.cc158 LogicalResult matchAndRewrite(GPUModuleOp op, in matchAndRewrite() function in tensorflow::GpuModuleOpLowering
171 LogicalResult matchAndRewrite(TerminatorOp op, in matchAndRewrite() function in tensorflow::TerminatorOpLowering
190 LogicalResult matchAndRewrite(IoFeedOp op, in matchAndRewrite() function in tensorflow::IoFeedOpLowering
247 LogicalResult matchAndRewrite(MemcpyOp op, in matchAndRewrite() function in tensorflow::MemcpyOpLowering
292 LogicalResult matchAndRewrite(MemsetOp op, in matchAndRewrite() function in tensorflow::MemsetOpLowering
330 LogicalResult matchAndRewrite(LaunchFuncOp op, in matchAndRewrite() function in tensorflow::LaunchFuncOpLowering
410 LogicalResult matchAndRewrite(GEMMOp op, in matchAndRewrite() function in tensorflow::GemmOpLowering
470 LogicalResult matchAndRewrite(CublasLtMatmulOp op, in matchAndRewrite() function in tensorflow::CublasLtMatmulOpLowering
568 LogicalResult matchAndRewrite(Conv op, in matchAndRewrite() function in tensorflow::ConvOpLowering
687 LogicalResult matchAndRewrite(WhileOp op, in matchAndRewrite() function in tensorflow::WhileOpLowering
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1019 LogicalResult matchAndRewrite(TF::BiasAddOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertBiasAddOp
1252 LogicalResult matchAndRewrite(OpT op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertConvDynamic
1281 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertConvOp
1420 LogicalResult matchAndRewrite(TF::PadV2Op op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertPadOpDynamic
1487 LogicalResult matchAndRewrite(TF::GatherNdOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertGatherNdOpDynamic
1589 LogicalResult matchAndRewrite(TF::FloorDivOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertBF16FloorDivOp
1617 LogicalResult matchAndRewrite(TF::BroadcastToOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertBroadcastToOp
1648 LogicalResult matchAndRewrite(TF::RollOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertRollOp
1704 LogicalResult matchAndRewrite(TF::LeakyReluOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertLeakyReluOp
1732 LogicalResult matchAndRewrite(TF::LeakyReluGradOp op, in matchAndRewrite() function in mlir::mhlo::__anond9a709240111::ConvertLeakyReluGradOp
[all …]

1234567