Home
last modified time | relevance | path

Searched refs:AddTensorflowProducerVersion (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_jitrt_pipeline.cc49 struct AddTensorflowProducerVersion struct
50 : public mlir::PassWrapper<AddTensorflowProducerVersion,
52 MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(AddTensorflowProducerVersion) in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() argument
135 pm.addPass(std::make_unique<AddTensorflowProducerVersion>()); in CreateTfJitRtPipeline()
263 pm.addPass(std::make_unique<AddTensorflowProducerVersion>()); in CreateJitRtSpecializationPipeline()
H A Dtf_cpurt_passes.cc572 struct AddTensorflowProducerVersion struct
573 : public mlir::PassWrapper<AddTensorflowProducerVersion,
648 void AddTensorflowProducerVersion::runOnOperation() { in runOnOperation()
748 pm.addPass(std::make_unique<AddTensorflowProducerVersion>()); in CreateTfCpuRtPipeline()