| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 55 LogicalResult matchAndRewrite(Operation* op, \
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
[all …]
|
| H A D | legalize_tfl.cc | 77 LogicalResult matchAndRewrite(Operation* op,
85 LogicalResult matchAndRewrite(Operation* op, \
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
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | device_transform_patterns.h | 36 LogicalResult matchAndRewrite(TFL::PackOp pack_op,
43 LogicalResult matchAndRewrite(TFL::SquaredDifferenceOp squared_diff_op,
51 LogicalResult matchAndRewrite(TFL::SplitOp split_op,
59 LogicalResult matchAndRewrite(TFL::SplitVOp splitv_op,
67 LogicalResult matchAndRewrite(TFL::Conv2DOp conv_op,
75 LogicalResult matchAndRewrite(TFL::SliceOp slice_op,
83 LogicalResult matchAndRewrite(TFL::FullyConnectedOp fc_op,
91 LogicalResult matchAndRewrite(TFL::ConcatenationOp concat_op,
99 LogicalResult matchAndRewrite(TFL::MeanOp mean_op,
107 LogicalResult matchAndRewrite(TFL::MeanOp mean_op,
|
| 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/lite/transforms/ |
| H A D | legalize_tf.cc | 163 LogicalResult matchAndRewrite(Operation* op, \
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
[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
|
| 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 …]
|
| /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/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 …]
|
| /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/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_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 | 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_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 …]
|
| /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/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/core/transforms/functional_to_region/ |
| H A D | impl.cc | 106 LogicalResult matchAndRewrite(IfLikeOp op,
122 LogicalResult matchAndRewrite(CaseLikeOp op,
139 LogicalResult matchAndRewrite(WhileLikeOp op,
155 LogicalResult matchAndRewrite(tfg::ForOp op,
315 LogicalResult ConvertIfLikeOp<IfLikeOp, IfLikeRegionOp>::matchAndRewrite( in matchAndRewrite() function in mlir::tfg::ConvertIfLikeOp
365 LogicalResult ConvertCaseLikeOp<CaseLikeOp, CaseLikeRegionOp>::matchAndRewrite( in matchAndRewrite() function in mlir::tfg::ConvertCaseLikeOp
430 ConvertWhileLikeOp<WhileLikeOp, WhileLikeRegionOp>::matchAndRewrite( in matchAndRewrite() function in mlir::tfg::ConvertWhileLikeOp
473 LogicalResult ConvertForOp::matchAndRewrite(tfg::ForOp op, in matchAndRewrite() function in mlir::tfg::ConvertForOp
|
| /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/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/mlir/quantization/tensorflow/utils/ |
| H A D | fake_quant_utils.cc | 48 .matchAndRewrite(fake_quant, builder); in ConvertFakeQuantOps()
52 .matchAndRewrite(fake_quant, builder); in ConvertFakeQuantOps()
57 .matchAndRewrite(fake_quant, builder); in ConvertFakeQuantOps()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | fake_quant_utils.cc | 84 .matchAndRewrite(fake_quant, builder); in ConvertFakeQuantOps()
88 .matchAndRewrite(fake_quant, builder); in ConvertFakeQuantOps()
93 .matchAndRewrite(fake_quant, builder); in ConvertFakeQuantOps()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_symbolic_shape_optimization.cc | 85 LogicalResult matchAndRewrite(shape::BroadcastOp op,
153 LogicalResult BroadcastOpLowering::matchAndRewrite( in matchAndRewrite() function in tensorflow::__anondb63220c0111::BroadcastOpLowering
174 LogicalResult matchAndRewrite(mhlo::DynamicBroadcastInDimOp op,
214 LogicalResult DynamicBroadcastInDimOpLowering::matchAndRewrite( in matchAndRewrite() function in tensorflow::__anondb63220c0111::DynamicBroadcastInDimOpLowering
|