Home
last modified time | relevance | path

Searched defs:jit_executable (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_kernels.cc168 Expected<JitExecutable> jit_executable = in CompileImpl() local
310 static void ExecuteImpl(JitExecutable& jit_executable, in ExecuteImpl()
375 Expected<AsyncValuePtr<JitExecutable>> jit_executable = in ExecuteImpl() local
H A Dtf_jitrt_kernels.cc490 Expected<JitExecutable> jit_executable = JitExecutable::Instantiate( in CompileImpl() local
703 static void ExecuteImpl(JitExecutable& jit_executable, in ExecuteImpl()
790 Expected<AsyncValuePtr<JitExecutable>> jit_executable = in ExecuteImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
H A Dtf_cpurt_executor.cc80 llvm::Expected<JitExecutable> jit_executable = in Compile() local
255 JitExecutable& jit_executable = it->getSecond(); in Execute() local
H A Dtf_jitrt_executor.cc112 llvm::Expected<JitExecutable> jit_executable = in Compile() local
199 JitExecutable& jit_executable = it->getSecond(); in Execute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/
H A Dtf_jitrt_benchmark_test.cc109 llvm::Expected<JitExecutable> jit_executable = in BM_InstantiateExecutable() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dmatmul_op_benchmark.h59 JitExecutable& jit_executable = in RunMatMulMlirBenchmark() local
H A Dbenchmark.cc92 llvm::Expected<JitExecutable> jit_executable = in CreateJitExecutable() local
H A Dbenchmark_mlir_function.cc95 JitExecutable& jit_executable = in RunJitRtBenchmark() local
H A Dcwise_op_unary_benchmark.h71 JitExecutable& jit_executable = CreateJitExecutable( in PrepareUnaryMlirBenchmark() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc135 auto jit_executable = runtime::JitExecutable::Instantiate( in Create() local
180 std::unique_ptr<runtime::JitExecutable> jit_executable, in JitRtExecutable()
H A Dgpu_compiler.cc1486 auto jit_executable = runtime::JitExecutable::Instantiate( in CompileAheadOfTime() local