| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 143 LogicalResult ConvertTFReluOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFReluOp
161 LogicalResult ConvertTFRelu6Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFRelu6Op
177 LogicalResult ConvertTFEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFEqualOp
191 LogicalResult ConvertTFNotEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFNotEqualOp
212 LogicalResult ConvertTFGreaterOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFGreaterOp
226 LogicalResult ConvertTFGreaterEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFGreaterEqualOp
241 LogicalResult ConvertTFAddOp::matchAndRewrite(Operation* op, in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFAddOp
255 LogicalResult ConvertTFAddV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFAddV2Op
270 LogicalResult ConvertTFAddNOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFAddNOp
295 LogicalResult ConvertTFSubOp::matchAndRewrite(Operation* op, in matchAndRewrite() function in mlir::tosa::__anon428c32b60111::ConvertTFSubOp
[all …]
|
| H A D | legalize_tfl.cc | 177 LogicalResult ConvertTFLReluOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLReluOp
227 LogicalResult ConvertTFLRelu1Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLRelu1Op
282 LogicalResult ConvertTFLRelu6Op::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLRelu6Op
390 LogicalResult ConvertTFLEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLEqualOp
403 LogicalResult ConvertTFLNotEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLNotEqualOp
420 LogicalResult ConvertTFLGreaterOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLGreaterOp
433 LogicalResult ConvertTFLGreaterEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLGreaterEqualOp
446 LogicalResult ConvertTFLLessOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLLessOp
458 LogicalResult ConvertTFLLessEqualOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLLessEqualOp
575 LogicalResult ConvertTFLAddOp::matchAndRewrite( in matchAndRewrite() function in mlir::tosa::__anon9433a7e20111::ConvertTFLAddOp
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 691 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 …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | legalize_tf.cc | 204 LogicalResult ConvertTFConcatV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFConcatV2Op
226 LogicalResult ConvertTFMatMulOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFMatMulOp
274 LogicalResult ConvertTFPackOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFPackOp
289 LogicalResult ConvertTFSplitOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFSplitOp
302 LogicalResult ConvertTFSplitVOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFSplitVOp
315 LogicalResult ConvertTFUnpackOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFUnpackOp
329 LogicalResult ConvertTFConv3DOp::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFConv3DOp
369 LogicalResult ConvertTFConv3DBackpropInputV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFConv3DBackpropInputV2Op
476 LogicalResult ConvertTFMatrixDiagV2Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFMatrixDiagV2Op
483 LogicalResult ConvertTFMatrixDiagV3Op::matchAndRewrite( in matchAndRewrite() function in mlir::TFL::__anon419a2d0d0111::ConvertTFMatrixDiagV3Op
[all …]
|
| H A D | lower_static_tensor_list.cc | 246 LogicalResult matchAndRewrite( in matchAndRewrite() function
320 LogicalResult matchAndRewrite( in matchAndRewrite() function
459 LogicalResult matchAndRewrite( in matchAndRewrite() function
677 LogicalResult matchAndRewrite( in matchAndRewrite() function
718 LogicalResult matchAndRewrite( in matchAndRewrite() function
877 LogicalResult matchAndRewrite( in matchAndRewrite() function
892 LogicalResult matchAndRewrite( in matchAndRewrite() function
912 LogicalResult matchAndRewrite( in matchAndRewrite() function
960 LogicalResult matchAndRewrite( in matchAndRewrite() function
1021 LogicalResult matchAndRewrite( in matchAndRewrite() function
[all …]
|
| H A D | optimize.cc | 556 LogicalResult matchAndRewrite(TFL::AddOp add_op, in matchAndRewrite() function
660 LogicalResult matchAndRewrite(TFL::FullyConnectedOp fc_op, in matchAndRewrite() function
730 LogicalResult matchAndRewrite(TFL::FullyConnectedOp fc_op, in matchAndRewrite() function
782 LogicalResult matchAndRewrite(ReluXOp relu_op, in matchAndRewrite() function
817 LogicalResult matchAndRewrite(TFL::MulOp mul_op, in matchAndRewrite() function
922 LogicalResult matchAndRewrite(TFL::MulOp mul_op, in matchAndRewrite() function
1007 LogicalResult matchAndRewrite(AffineOpType fc_op, in matchAndRewrite() function
1159 LogicalResult matchAndRewrite(BinaryOpType binary_op, in matchAndRewrite() function
1285 LogicalResult matchAndRewrite(TFL::TransposeOp transpose_op, in matchAndRewrite() function
1361 LogicalResult matchAndRewrite(TFL::FullyConnectedOp fully_connected_op, in matchAndRewrite() function
[all …]
|
| H A D | legalize_hashtables.cc | 53 LogicalResult matchAndRewrite(TF::HashTableV2Op hashtable_op, in matchAndRewrite() function in mlir::TFL::__anon1b1670b20111::LegalizeHashTableOpPattern
77 LogicalResult matchAndRewrite(TF::LookupTableFindV2Op find_op, in matchAndRewrite() function in mlir::TFL::__anon1b1670b20111::LegalizeHashTableFindOpPattern
95 LogicalResult matchAndRewrite(TF::LookupTableImportV2Op import_op, in matchAndRewrite() function in mlir::TFL::__anon1b1670b20111::LegalizeHashTableImportOpPattern
113 LogicalResult matchAndRewrite(TF::LookupTableSizeV2Op size_op, in matchAndRewrite() function in mlir::TFL::__anon1b1670b20111::LegalizeHashTableSizeOpPattern
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tfg-to-tfe.cc | 139 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertGraphOp
174 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertGraphFuncOp
243 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertReturnOp
257 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertControlTriggerOp
277 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertEnterOp
297 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertExitOp
317 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertLoopCondOp
337 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertMergeOp
357 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertSwitchOp
377 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anon178d5bc90111::ConvertSwitchNOp
[all …]
|
| H A D | lower_tf.cc | 221 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerAddNOp
286 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerDynamicStitchOp
370 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::ConvertFakeQuantWithMinMaxVarsOp
497 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerInvertPermutationOp
578 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerLgammaOp
773 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerPackOp
849 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerSpaceToBatchNDOp
1058 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerBatchToSpaceND
1230 LogicalResult matchAndRewrite(Operation *src_op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::LowerSparseMatMulOp
1274 LogicalResult matchAndRewrite(_UnaryOpsCompositionOp op, in matchAndRewrite() function in mlir::TF::__anon9aef87ee0111::Lower_UnaryOpsComposition
[all …]
|
| H A D | legalize_hlo.cc | 200 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::Convert1DConvOp
397 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::Convert2DConvOp
706 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertNonTrivialConvOp
1021 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertSliceOp
1040 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertDynamicSliceOp
1196 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertDynamicUpdateSliceOp
1571 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertSortToTfTopk
1873 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertReduceOpToTfOp
2041 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertReduceOpToTfArgMinMax
2237 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::TF::__anonab43b57b0111::ConvertIotaOpToTfRange
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 1019 LogicalResult matchAndRewrite(TF::BiasAddOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertBiasAddOp
1252 LogicalResult matchAndRewrite(OpT op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertConvDynamic
1281 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertConvOp
1420 LogicalResult matchAndRewrite(TF::PadV2Op op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertPadOpDynamic
1487 LogicalResult matchAndRewrite(TF::GatherNdOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertGatherNdOpDynamic
1589 LogicalResult matchAndRewrite(TF::FloorDivOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertBF16FloorDivOp
1617 LogicalResult matchAndRewrite(TF::BroadcastToOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertBroadcastToOp
1648 LogicalResult matchAndRewrite(TF::RollOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertRollOp
1704 LogicalResult matchAndRewrite(TF::LeakyReluOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertLeakyReluOp
1732 LogicalResult matchAndRewrite(TF::LeakyReluGradOp op, in matchAndRewrite() function in mlir::mhlo::__anon6d04cea40111::ConvertLeakyReluGradOp
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| H A D | bufferize_tiled_loop.cc | 119 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | embed_tf_framework.cc | 36 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anona0bbadb60111::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
|
| H A D | tf_framework_legalize_to_llvm.cc | 112 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::TFAllocOpConverter
226 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::TFDeallocOpConverter
259 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::JITCompileFromStrOpConverter
322 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::JITExecuteOpConverter
405 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::ReportErrorOpConverter
460 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::NullContextOpConverter
472 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::NullMemRefOpConverter
551 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::kernel_gen::tf_framework::__anonbe477fa20111::IsValidMemRefOpConverter
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | lmhlo_to_jitrt.cc | 158 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 …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | merge_assuming_ops.cc | 52 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
|
| H A D | legalize_to_linalg.cc | 203 LogicalResult matchAndRewrite( in matchAndRewrite() function
381 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::EinsumToLinalgConverter
572 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::ScalarPointwiseToStandardConverter
604 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::DataMovementOpConverter
730 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::HloDynamicBroadcastInDimConverter
838 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::RealDynamicSliceConverter
926 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::ReshapeOpConverter
1034 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::IotaConverter
1086 LogicalResult matchAndRewrite( in matchAndRewrite() function
1175 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anon071fe3ef0111::ConstConverterTensor
[all …]
|
| H A D | chlo_legalize_to_hlo.cc | 52 LogicalResult matchAndRewrite( in matchAndRewrite() function
233 LogicalResult matchAndRewrite( in matchAndRewrite() function
586 LogicalResult matchAndRewrite( in matchAndRewrite() function
612 LogicalResult matchAndRewrite( in matchAndRewrite() function
820 LogicalResult matchAndRewrite( in matchAndRewrite() function
1129 LogicalResult matchAndRewrite( in matchAndRewrite() function
1142 LogicalResult matchAndRewrite( in matchAndRewrite() function
1240 LogicalResult matchAndRewrite( in matchAndRewrite() function
1251 LogicalResult matchAndRewrite( in matchAndRewrite() function
1323 LogicalResult matchAndRewrite( in matchAndRewrite() function
[all …]
|
| H A D | hlo_legalize_to_lhlo.cc | 144 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anonc86515690111::HloToLhloOpConverter
167 LogicalResult matchAndRewrite( in matchAndRewrite() function in mlir::mhlo::__anonc86515690111::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
|
| H A D | legalize_shape_computations.cc | 76 LogicalResult matchAndRewrite(OpTy op, in matchAndRewrite() function in mlir::__anonb70784cc0111::MhloElementwiseConverter
114 LogicalResult matchAndRewrite(mhlo::ConcatenateOp op, in matchAndRewrite() function in mlir::__anonb70784cc0111::ConcatenateConverter
147 LogicalResult matchAndRewrite(mhlo::GetDimensionSizeOp op, in matchAndRewrite() function in mlir::__anonb70784cc0111::GetDimSizeConverter
169 LogicalResult matchAndRewrite(mhlo::ReshapeOp op, in matchAndRewrite() function in mlir::__anonb70784cc0111::ReshapeConverter
|
| H A D | legalize_mhlo_to_thlo.cc | 50 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | device_transform_patterns.cc | 206 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt.cc | 128 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::FallbackExecuteOpConversion
353 mlir::LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::FallbackConstOpConversion
387 mlir::LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::FallbackSetResourceOp
430 mlir::LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::FallbackGetResourceOp
486 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::TFDeviceRemoteRunOpConversion
563 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::FallbackBatchFunctionOpConversion
626 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::CoreRTConstDenseTensorOpConversion
667 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::TFRTFuncOpSignatureConversion
749 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::CoreRTConstStringTensorOpConversion
812 LogicalResult matchAndRewrite( in matchAndRewrite() function in tensorflow::__anon42f45b140111::CoreRTExecuteOpConversion
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| H A D | tfr_ops.cc | 386 LogicalResult matchAndRewrite(ConstantTensorOp cst_tensor_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::ConvertConstToTensorConst
437 LogicalResult matchAndRewrite(CastOp cast_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveRedundantCast
500 LogicalResult matchAndRewrite(GetShapeOp shape_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::GetTensorShape
516 LogicalResult matchAndRewrite(GetElementOp ge_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveRedundantGetElement
543 LogicalResult matchAndRewrite(GetLengthOp gl_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveRedundantGetLength
561 LogicalResult matchAndRewrite(BuildListOp bl_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::BuildConstantListAsAttr
592 LogicalResult matchAndRewrite(TFRQuantRawDataOp raw_data_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveRawDataOp
649 LogicalResult matchAndRewrite(TFRQuantQParamsOp qparams_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveQParamsOp
720 LogicalResult matchAndRewrite(TFRQuantScaleFactorOp scale_factor_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveScaleFactorOp
788 LogicalResult matchAndRewrite(TFRQuantRescaleOp rescale_op, in matchAndRewrite() function in mlir::TFR::__anoncf4f53ce0411::RemoveRescaleOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/ |
| H A D | pass.cc | 227 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::__anon17f873fd0311::ReifyOperationOutputShapes
315 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::__anon17f873fd0311::DropAttributes
504 LogicalResult matchAndRewrite(IfLikeOp op, in matchAndRewrite() function
526 LogicalResult matchAndRewrite(CaseLikeOp op, in matchAndRewrite() function
544 LogicalResult matchAndRewrite(WhileOrForLikeOp op, in matchAndRewrite() function
563 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite() function in mlir::tfg::__anon17f873fd0711::MaterializeOutputShapesBase
|