Home
last modified time | relevance | path

Searched refs:create_specialization_pipeline (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/
H A Dtf_jitrt_benchmark_test.cc99 opts.compiler.create_specialization_pipeline = [&](mlir::PassManager& pm) { in BM_InstantiateExecutable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
H A Djit_compiler.h63 std::function<void(mlir::PassManager&)> create_specialization_pipeline; member
H A Djit_compiler.cc116 return RunPipeline(module, opts.create_specialization_pipeline); in RunSpecializationPipeline()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dbenchmark.cc75 opts.compiler.create_specialization_pipeline = in CreateJitExecutable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
H A Dtf_jitrt_executor.cc104 opts.compiler.create_specialization_pipeline = in Compile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_jitrt_kernels.cc475 opts.compiler.create_specialization_pipeline = [=](mlir::PassManager& pm) { in CompileImpl()