Home
last modified time | relevance | path

Searched refs:getCanonicalizationPatterns (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dbroadcast_propagation.cc437 mhlo::DynamicBroadcastInDimOp::getCanonicalizationPatterns(*patterns, in PopulateBroadcastsPropagationPatterns()
439 mhlo::DynamicReshapeOp::getCanonicalizationPatterns(*patterns, context); in PopulateBroadcastsPropagationPatterns()
440 shape::AssumingAllOp::getCanonicalizationPatterns(*patterns, context); in PopulateBroadcastsPropagationPatterns()
441 shape::AssumingOp::getCanonicalizationPatterns(*patterns, context); in PopulateBroadcastsPropagationPatterns()
442 shape::BroadcastOp::getCanonicalizationPatterns(*patterns, context); in PopulateBroadcastsPropagationPatterns()
443 shape::CstrBroadcastableOp::getCanonicalizationPatterns(*patterns, context); in PopulateBroadcastsPropagationPatterns()
444 tensor::CastOp::getCanonicalizationPatterns(*patterns, context); in PopulateBroadcastsPropagationPatterns()
Drank_specialization.cc942 shape::BroadcastOp::getCanonicalizationPatterns(*patterns, context); in PopulateRankSpecializationToSCFPatterns()
943 shape::ShapeOfOp::getCanonicalizationPatterns(*patterns, context); in PopulateRankSpecializationToSCFPatterns()
944 shape::AnyOp::getCanonicalizationPatterns(*patterns, context); in PopulateRankSpecializationToSCFPatterns()
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc777 void ConstantTensorOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::ConstantTensorOp
782 void CastOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFR::CastOp
787 void GetShapeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFR::GetShapeOp
792 void GetElementOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::GetElementOp
797 void GetLengthOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFR::GetLengthOp
802 void BuildListOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFR::BuildListOp
807 void TFRQuantRawDataOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantRawDataOp
812 void TFRQuantQParamsOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantQParamsOp
817 void TFRQuantRescaleOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantRescaleOp
822 void TFRQuantScaleFactorOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFR::TFRQuantScaleFactorOp
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc87 void AddOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::AddOp
105 void AddV2Op::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::AddV2Op
156 void AssertOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::AssertOp
261 void BatchMatMulOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::BatchMatMulOp
266 void BatchMatMulV2Op::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::BatchMatMulV2Op
414 void BatchToSpaceOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::BatchToSpaceOp
537 void BiasAddV1Op::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::BiasAddV1Op
546 void BitcastOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::BitcastOp
752 void CaseOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::CaseOp
923 void CaseRegionOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::CaseRegionOp
[all …]
Dtf_ops_n_z.cc345 void PackOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::PackOp
498 void QuantizeAndDequantizeV2Op::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::QuantizeAndDequantizeV2Op
529 void ReadVariableOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::ReadVariableOp
691 void RealDivOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::RealDivOp
843 void ReshapeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::ReshapeOp
1137 void ShapeNOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::ShapeNOp
1574 void SquareOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::SquareOp
1583 void SubOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::SubOp
2449 void ToBoolOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TF::ToBoolOp
2602 void TruncateDivOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TF::TruncateDivOp
[all …]
Dtf_executor.cc1018 void GraphOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::tf_executor::GraphOp
1076 void IslandOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::tf_executor::IslandOp
1106 void ControlTriggerOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::tf_executor::ControlTriggerOp
Dtf_device.cc758 void ClusterOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::tf_device::ClusterOp
787 void LaunchOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::tf_device::LaunchOp
Dtf_saved_model.cc493 void SessionInitializerOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::tf_saved_model::SessionInitializerOp
/external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/
Druntime_fallback_combine.cc77 void mlir::tfd::ConvertTftToDhtOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::tfd::ConvertTftToDhtOp
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dlhlo_ops.cc174 void ConstOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::lmhlo::ConstOp
279 void ReduceOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::lmhlo::ReduceOp
Dhlo_ops.cc337 void GatherOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::mhlo::GatherOp
548 void IotaOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::mhlo::IotaOp
638 void DynamicIotaOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::DynamicIotaOp
781 void ConvertOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::mhlo::ConvertOp
908 void TupleOp::getCanonicalizationPatterns(OwningRewritePatternList& results, in getCanonicalizationPatterns() function in mlir::mhlo::TupleOp
1219 void DynamicBroadcastInDimOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::DynamicBroadcastInDimOp
1485 void ConcatenateOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::ConcatenateOp
1760 void DynamicReshapeOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::DynamicReshapeOp
1819 void DynamicSliceOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::DynamicSliceOp
1935 void RealDynamicSliceOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::mhlo::RealDynamicSliceOp
[all …]
Dchlo_ops.cc401 void ConstantLikeOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::chlo::ConstantLikeOp
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Dcanonicalize.cc168 op->getAbstractOperation()->getCanonicalizationPatterns(patterns, in populateCanonicalizationPatterns()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1022 void FullyConnectedOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFL::FullyConnectedOp
1040 void Conv2DOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::Conv2DOp
1162 void DepthwiseConv2DOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in mlir::TFL::DepthwiseConv2DOp
1602 void ReshapeOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::ReshapeOp
1820 void PackOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::PackOp
1956 void SliceOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::SliceOp
2061 void FakeQuantOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::FakeQuantOp
2404 void LSTMOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::LSTMOp
2730 void ConstOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::ConstOp
3371 void WhileOp::getCanonicalizationPatterns(OwningRewritePatternList &results, in getCanonicalizationPatterns() function in mlir::TFL::WhileOp
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dshape_simplification.cc253 op->getCanonicalizationPatterns(patterns, context); in runOnFunction()
/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
Dtfrt_fallback_async.cc338 void CoreRTTensorHandleToFallbackTensorOp::getCanonicalizationPatterns( in getCanonicalizationPatterns() function in tfrt::fallback_async::CoreRTTensorHandleToFallbackTensorOp
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc63 op->getCanonicalizationPatterns(*patterns, context); in AddCanonicalizationPatterns()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc1167 mhlo::GatherOp::getCanonicalizationPatterns(patterns, context); in ConvertTf2XlaOps()
Doptimize.cc1421 op->getCanonicalizationPatterns(*patterns, context); in AddCanonicalizationPatterns()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc7417 chlo::ConstantLikeOp::getCanonicalizationPatterns(patterns, context); in legalizeTF()