Home
last modified time | relevance | path

Searched refs:LocalExecutable (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_compilation_cache.h34 explicit XRTCompilationCacheEntry(xla::LocalExecutable* executable) in XRTCompilationCacheEntry()
38 xla::LocalExecutable* get_executable() const { return executable; } in get_executable()
41 xla::LocalExecutable* executable;
114 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
151 std::unique_ptr<xla::LocalExecutable> program;
208 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
Dxrt_compilation_cache.cc170 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>& in InitializeEntry()
191 std::unique_ptr<xla::LocalExecutable> program; in InitializeEntry()
216 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>& in CompileIfKeyAbsent()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.h78 xla::LocalExecutable** out_executable);
87 xla::LocalExecutable** out_executable);
132 xla::LocalExecutable** out_executable);
138 std::unique_ptr<xla::LocalExecutable>* executable);
161 std::unique_ptr<xla::LocalExecutable> executable GUARDED_BY(mu);
Dxla_compile_on_demand_op.h50 xla::LocalExecutable** executable);
53 xla::LocalExecutable* executable);
Dxla_compile_on_demand_op.cc54 xla::LocalExecutable* executable) { in Run()
118 xla::LocalExecutable** executable) { in Compile()
213 xla::LocalExecutable* executable; in Compute()
Dxla_compilation_cache.cc150 std::unique_ptr<xla::LocalExecutable>* executable) { in BuildExecutable()
180 xla::LocalExecutable** out_executable) { in Compile()
210 xla::LocalExecutable** out_executable) { in CompileSingleOp()
254 xla::LocalExecutable** out_executable) { in CompileImpl()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc44 LocalExecutable::LocalExecutable(std::unique_ptr<Executable> executable, in LocalExecutable() function in xla::LocalExecutable
54 Status LocalExecutable::ValidateExecutionOptions( in ValidateExecutionOptions()
119 LocalExecutable::RunHelper(const absl::Span<const Shape* const> argument_shapes, in RunHelper()
171 StatusOr<ScopedShapedBuffer> LocalExecutable::Run( in Run()
231 StatusOr<ScopedShapedBuffer> LocalExecutable::RunAsync( in RunAsync()
272 StatusOr<ExecutionOutput> LocalExecutable::RunAsync( in RunAsync()
340 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> LocalClient::Compile( in Compile()
354 std::vector<std::unique_ptr<LocalExecutable>> local_executables; in Compile()
358 local_executables.push_back(absl::make_unique<LocalExecutable>( in Compile()
Dlocal_client.h41 class LocalExecutable {
45 LocalExecutable(std::unique_ptr<Executable> executable, Backend* backend,
120 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> Compile(
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc66 std::unique_ptr<xla::LocalExecutable>* program);
84 std::unique_ptr<xla::LocalExecutable>* program) { in Compile()
168 key, &uid, [&](std::unique_ptr<xla::LocalExecutable>* program) { in Compute()
179 xla::LocalExecutable* executable = entry->get().get_executable(); in Compute()
Dxrt_execute_op.cc150 xla::LocalExecutable* executable, const InputBuffers& input_buffers, in RunExecutable()
198 xla::LocalExecutable* executable, const InputBuffers& input_buffers, in ExecuteComputation()
216 xla::LocalExecutable* executable, in ExecuteComputation()
295 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
379 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
70 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in TestWithDeviceCount()
Dlocal_client_execute_test.cc960 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in BM_LocalClientOverhead()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc107 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
121 xla::LocalExecutable* executable() const { return executable_; } in executable()
132 xla::LocalExecutable* executable_;
278 xla::LocalExecutable** executable) { in CompileToLocalExecutable()
348 xla::LocalExecutable* executable; in Compute()
485 xla::LocalExecutable* executable; in Compute()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc114 StatusOr<std::unique_ptr<LocalExecutable>> CompileExecutable( in CompileExecutable()
222 LocalExecutable* executable, in ReplayComputation()
442 std::vector<StatusOr<std::unique_ptr<LocalExecutable>>> executables; in RealMain()
472 LocalExecutable* executable = executables[i].ValueOrDie().get(); in RealMain()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.h65 std::unique_ptr<xla::LocalExecutable> executable_;
Dxla_jit_compiled_cpu_function.cc123 std::unique_ptr<xla::LocalExecutable> executable = std::move(executables[0]); in Compile()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_client.h308 PyLocalExecutable(std::vector<std::unique_ptr<LocalExecutable>> executables,
368 std::vector<std::shared_ptr<LocalExecutable>> executables_;
Dlocal_client.cc673 std::vector<std::unique_ptr<LocalExecutable>> executables, in PyLocalExecutable()
1097 std::vector<std::unique_ptr<LocalExecutable>> local_executables, in Compile()
Dxla_client.py152 return _xla.LocalExecutable.Compile(c_computation,
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.h214 friend class LocalExecutable;
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc288 std::vector<std::unique_ptr<xla::LocalExecutable>> local_executables = in XlaCompiledProgramShape()
292 std::unique_ptr<xla::LocalExecutable> local_executable = in XlaCompiledProgramShape()