Home
last modified time | relevance | path

Searched defs:getCanonicalizationPatterns (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc92 void AddOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::AddOp
147 void AddV2Op::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::AddV2Op
200 void AssertOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::AssertOp
308 void BatchMatMulOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::BatchMatMulOp
313 void BatchMatMulV2Op::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::BatchMatMulV2Op
462 void BatchToSpaceOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::BatchToSpaceOp
588 void BiasAddV1Op::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::BiasAddV1Op
597 void BitcastOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::BitcastOp
804 void CaseOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::CaseOp
979 void CaseRegionOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::CaseRegionOp
[all …]
H A Dtf_ops_n_z.cc362 void PackOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::PackOp
526 void QuantizeAndDequantizeV2Op::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::QuantizeAndDequantizeV2Op
558 void ReadVariableOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::ReadVariableOp
721 void RealDivOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::RealDivOp
734 void ReluOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::ReluOp
883 void ReshapeOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::ReshapeOp
1186 void ShapeNOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::ShapeNOp
1634 void SquareOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::SquareOp
1667 void SubOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TF::SubOp
2341 void TensorListGetItemOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::TensorListGetItemOp
[all …]
H A Dtf_executor.cc987 void GraphOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::tf_executor::GraphOp
1045 void IslandOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::tf_executor::IslandOp
1075 void ControlTriggerOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::tf_executor::ControlTriggerOp
H A Dtf_device.cc766 void ClusterOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::tf_device::ClusterOp
795 void LaunchOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::tf_device::LaunchOp
H A Dtf_saved_model.cc495 void SessionInitializerOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::tf_saved_model::SessionInitializerOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
H A Dtfr_ops.cc845 void ConstantTensorOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::ConstantTensorOp
850 void CastOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::CastOp
855 void GetShapeOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::GetShapeOp
860 void GetElementOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::GetElementOp
865 void GetLengthOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::GetLengthOp
870 void BuildListOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::BuildListOp
875 void TFRQuantRawDataOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantRawDataOp
880 void TFRQuantQParamsOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantQParamsOp
885 void TFRQuantRescaleOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantRescaleOp
890 void TFRQuantScaleFactorOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantScaleFactorOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1010 void BroadcastToOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::BroadcastToOp
1164 void FullyConnectedOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::FullyConnectedOp
1182 void Conv2DOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::Conv2DOp
1304 void DepthwiseConv2DOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::DepthwiseConv2DOp
1747 void ReshapeOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::ReshapeOp
1999 void PackOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::PackOp
2134 void SliceOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::SliceOp
2239 void FakeQuantOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::FakeQuantOp
2587 void LSTMOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::LSTMOp
2933 void ConstOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::TFL::ConstOp
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1179 void DotGeneralOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::DotGeneralOp
1402 void GatherOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::GatherOp
1905 void IotaOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::IotaOp
2046 void DynamicIotaOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::DynamicIotaOp
2480 void ConvolutionOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::ConvolutionOp
2638 void ConvertOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::ConvertOp
2728 void TupleOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::TupleOp
3325 void BroadcastInDimOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::BroadcastInDimOp
3502 void DynamicBroadcastInDimOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::DynamicBroadcastInDimOp
3822 void ConcatenateOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::mhlo::ConcatenateOp
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/IR/
H A Dlhlo_ops.cc211 void ConstantOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::lmhlo::ConstantOp
378 void ReduceOp::getCanonicalizationPatterns(RewritePatternSet& results, in getCanonicalizationPatterns() function in mlir::lmhlo::ReduceOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/
H A Druntime_fallback_combine.cc77 void mlir::tfd::ConvertTftToDhtOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::tfd::ConvertTftToDhtOp
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
H A Dtfrt_fallback_async.cc338 void CoreRTTensorHandleToFallbackTensorOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in tfrt::fallback_async::CoreRTTensorHandleToFallbackTensorOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
H A Dtfrt_fallback_async.cc432 void CoreRTTensorHandleToFallbackTensorOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in tfrt::fallback_async::CoreRTTensorHandleToFallbackTensorOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
H A Dgml_st_ops.cc1163 void LoopOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns() function in mlir::gml_st::LoopOp