| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt_kernels.cc | 168 Expected<JitExecutable> jit_executable = in CompileImpl() local
172 if (auto err = jit_executable.takeError()) in CompileImpl()
175 ptr.emplace(std::move(*jit_executable)); in CompileImpl()
310 static void ExecuteImpl(JitExecutable& jit_executable, in ExecuteImpl() argument
321 AsyncValuePtr<Executable> executable = jit_executable.GetExecutable( in ExecuteImpl()
375 Expected<AsyncValuePtr<JitExecutable>> jit_executable = in ExecuteImpl() local
378 if (auto err = jit_executable.takeError()) in ExecuteImpl()
382 if (jit_executable->IsAvailable()) { in ExecuteImpl()
383 if (jit_executable->IsError()) { in ExecuteImpl()
384 EmitErrors(results, jit_executable->GetError(), exec_ctx); in ExecuteImpl()
[all …]
|
| D | tf_jitrt_kernels.cc | 490 Expected<JitExecutable> jit_executable = JitExecutable::Instantiate( in CompileImpl() local
508 if (auto err = jit_executable.takeError()) in CompileImpl()
511 ref.emplace(std::move(*jit_executable)); in CompileImpl()
703 static void ExecuteImpl(JitExecutable& jit_executable, in ExecuteImpl() argument
716 Expected<AsyncValuePtr<Executable>> executable = jit_executable.GetExecutable( in ExecuteImpl()
790 Expected<AsyncValuePtr<JitExecutable>> jit_executable = in ExecuteImpl() local
793 if (LLVM_UNLIKELY(!jit_executable)) in ExecuteImpl()
794 return ReturnErrors(results, jit_executable.takeError(), exec_ctx); in ExecuteImpl()
797 if (LLVM_LIKELY(jit_executable->IsConcrete())) in ExecuteImpl()
798 return ExecuteImpl(**jit_executable, operands, results, exec_ctx, debug); in ExecuteImpl()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/ |
| D | tf_jitrt_executor.cc | 112 llvm::Expected<JitExecutable> jit_executable = in Compile() local
114 if (auto err = jit_executable.takeError()) in Compile()
119 jit_executables_.insert({hdl, std::move(*jit_executable)}); in Compile()
199 JitExecutable& jit_executable = it->getSecond(); in Execute() local
215 jit_executable.GetExecutable(memrefs); in Execute()
|
| D | tf_cpurt_executor.cc | 80 llvm::Expected<JitExecutable> jit_executable = in Compile() local
82 if (auto err = jit_executable.takeError()) in Compile()
87 jit_executables_.insert({hdl, std::move(*jit_executable)}); in Compile()
255 JitExecutable& jit_executable = it->getSecond(); in Execute() local
270 jit_executable.GetExecutable(memrefs, exec_ctx); in Execute()
|
| D | BUILD | 47 "//tensorflow/compiler/xla/runtime:jit_executable",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | benchmark.cc | 92 llvm::Expected<JitExecutable> jit_executable = in CreateJitExecutable() local
94 if (auto err = jit_executable.takeError()) in CreateJitExecutable()
98 auto storage = std::make_unique<JitExecutable>(std::move(*jit_executable)); in CreateJitExecutable()
|
| D | benchmark_mlir_function.cc | 95 JitExecutable& jit_executable = in RunJitRtBenchmark() local
118 jit_executable.GetExecutable(operands); in RunJitRtBenchmark()
|
| D | matmul_op_benchmark.h | 59 JitExecutable& jit_executable = in RunMatMulMlirBenchmark() local
99 jit_executable.GetExecutable(operands); in RunMatMulMlirBenchmark()
|
| D | cwise_op_unary_benchmark.h | 71 JitExecutable& jit_executable = CreateJitExecutable( in PrepareUnaryMlirBenchmark() local
95 jit_executable.GetExecutable(operands); in PrepareUnaryMlirBenchmark()
|
| D | BUILD | 21 "//tensorflow/compiler/xla/runtime:jit_executable",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/ |
| D | tf_jitrt_benchmark_test.cc | 109 llvm::Expected<JitExecutable> jit_executable = in BM_InstantiateExecutable() local
111 if (auto err = jit_executable.takeError()) in BM_InstantiateExecutable()
|
| D | BUILD | 33 "//tensorflow/compiler/xla/runtime:jit_executable",
|
| /external/tensorflow/tensorflow/compiler/xla/runtime/ |
| D | BUILD | 192 name = "jit_executable",
193 srcs = ["jit_executable.cc"],
194 hdrs = ["jit_executable.h"],
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 135 auto jit_executable = runtime::JitExecutable::Instantiate( in Create() local
137 if (auto err = jit_executable.takeError()) in Create()
144 std::make_unique<runtime::JitExecutable>(std::move(*jit_executable)), in Create()
180 std::unique_ptr<runtime::JitExecutable> jit_executable, in JitRtExecutable() argument
183 jit_executable_(std::move(jit_executable)), in JitRtExecutable()
|
| D | gpu_compiler.cc | 1486 auto jit_executable = runtime::JitExecutable::Instantiate( in CompileAheadOfTime() local
1488 if (auto err = jit_executable.takeError()) in CompileAheadOfTime()
1493 runtime::Executable& executable = jit_executable->DefaultExecutable().get(); in CompileAheadOfTime()
|
| D | BUILD | 562 # "//tensorflow/compiler/xla/runtime:jit_executable",
710 "//tensorflow/compiler/xla/runtime:jit_executable",
1740 "//tensorflow/compiler/xla/runtime:jit_executable",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ |
| D | BUILD | 211 "//tensorflow/compiler/xla/runtime:jit_executable",
242 "//tensorflow/compiler/xla/runtime:jit_executable",
|