| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.cc | 45 LocalExecutable::LocalExecutable(std::unique_ptr<Executable> executable, in LocalExecutable() function in xla::LocalExecutable
55 Status LocalExecutable::ValidateExecutionOptions( in ValidateExecutionOptions()
120 LocalExecutable::RunHelper(const absl::Span<const Shape* const> argument_shapes, in RunHelper()
175 StatusOr<ScopedShapedBuffer> LocalExecutable::Run( in Run()
189 StatusOr<ExecutionOutput> LocalExecutable::Run( in Run()
243 StatusOr<ScopedShapedBuffer> LocalExecutable::RunAsync( in RunAsync()
283 StatusOr<ExecutionOutput> LocalExecutable::RunAsync( in RunAsync()
325 StatusOr<ExecutionOutput> LocalExecutable::RunAsync( in RunAsync()
387 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> LocalClient::Compile( in Compile()
397 std::vector<std::unique_ptr<LocalExecutable>> local_executables; in Compile()
[all …]
|
| H A D | local_client.h | 42 class LocalExecutable {
46 LocalExecutable(std::unique_ptr<Executable> executable, Backend* backend,
144 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> Compile(
159 StatusOr<std::unique_ptr<LocalExecutable>> Load(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_compilation_cache.h | 34 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>*)>&
152 std::unique_ptr<xla::LocalExecutable> program;
209 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
|
| H A D | xrt_compilation_cache.cc | 170 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_compilation_cache.h | 117 xla::LocalExecutable** out_executable);
128 xla::LocalExecutable** out_executable);
171 xla::LocalExecutable** out_executable);
177 std::unique_ptr<xla::LocalExecutable>* executable);
188 StatusOr<std::unique_ptr<xla::LocalExecutable>> LoadExecutable(
221 std::unique_ptr<xla::LocalExecutable> executable TF_GUARDED_BY(mu);
|
| H A D | get_compiler_ir.cc | 44 static StatusOr<std::unique_ptr<xla::LocalExecutable>> BuildExecutable( in BuildExecutable()
71 std::vector<std::unique_ptr<xla::LocalExecutable>> executables, in BuildExecutable()
188 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::LocalExecutable> executable, in GetCompilerIr()
198 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::LocalExecutable> executable, in GetCompilerIr()
204 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::LocalExecutable> executable, in GetCompilerIr()
|
| H A D | xla_compile_on_demand_op.cc | 50 xla::LocalExecutable* executable, in Run()
113 xla::LocalExecutable** executable) { in Compile()
173 xla::LocalExecutable* executable; in Compute()
|
| H A D | xla_compile_on_demand_op.h | 50 xla::LocalExecutable** executable);
54 xla::LocalExecutable* executable,
|
| H A D | xla_compilation_cache.cc | 281 std::unique_ptr<xla::LocalExecutable>* executable) { in BuildExecutable()
314 StatusOr<std::unique_ptr<xla::LocalExecutable>>
332 xla::LocalExecutable** out_executable) { in Compile()
454 xla::LocalExecutable** out_executable) { in CompileSingleOp()
540 StatusOr<std::unique_ptr<xla::LocalExecutable>> executable = LoadExecutable( in CompileStrict()
694 xla::LocalExecutable** out_executable) { in CompileImpl()
|
| H A D | xla_compilation_cache_disable_test.cc | 47 xla::LocalExecutable* executable; in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/ |
| H A D | xla_ops.cc | 80 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
93 xla::LocalExecutable* executable() const { return executable_; } in executable()
104 xla::LocalExecutable* executable_;
174 xla::ExecutableRunOptions run_options, xla::LocalExecutable* executable, in RunExecutable()
245 xla::LocalExecutable** executable) { in CompileToLocalExecutable()
312 xla::LocalExecutable* executable; in ComputeAsync()
483 xla::LocalExecutable* executable; in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_compile_ops.cc | 107 std::unique_ptr<xla::LocalExecutable>* program);
125 std::unique_ptr<xla::LocalExecutable>* program) { in Compile()
218 key, &uid, [&](std::unique_ptr<xla::LocalExecutable>* program) { in Compute()
229 xla::LocalExecutable* executable = entry->get().get_executable(); in Compute()
|
| H A D | xrt_execute_op.cc | 89 xla::LocalExecutable* executable, XRTMemoryManager::WorkingSet* working_set, in GetInputTuples()
299 xla::LocalExecutable* executable, in RunExecutable()
381 xla::LocalExecutable* executable, in ExecuteComputation()
401 xla::LocalExecutable* executable, in ExecuteComputation()
484 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
567 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | replay_computation.cc | 113 StatusOr<std::unique_ptr<LocalExecutable>> CompileExecutable( in CompileExecutable()
231 LocalExecutable* executable, in ReplayComputation()
470 std::vector<StatusOr<std::unique_ptr<LocalExecutable>>> executables; in RealMain()
500 LocalExecutable* executable = executables[i].ValueOrDie().get(); in RealMain()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
71 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in TestWithDeviceCount()
|
| H A D | local_client_execute_test.cc | 966 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in BM_LocalClientOverhead()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tpu_client.cc | 194 auto local_executable = std::make_unique<LocalExecutable>( in DeserializeExecutable()
197 std::vector<std::unique_ptr<LocalExecutable>> local_executables; in DeserializeExecutable()
|
| H A D | pjrt_stream_executor_client.h | 713 std::vector<std::unique_ptr<LocalExecutable>> executables,
786 absl::Span<const std::shared_ptr<LocalExecutable>> executables() const { in executables()
844 std::vector<std::shared_ptr<LocalExecutable>> executables_;
|
| H A D | pjrt_stream_executor_client.cc | 1762 std::vector<std::unique_ptr<LocalExecutable>> executables, in PjRtStreamExecutorExecutable()
2472 std::vector<std::unique_ptr<LocalExecutable>> local_executables, in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_jit_compiled_cpu_function.h | 65 std::unique_ptr<xla::LocalExecutable> executable_;
|
| H A D | xla_jit_compiled_cpu_function.cc | 123 std::unique_ptr<xla::LocalExecutable> executable = std::move(executables[0]); in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.h | 213 friend class LocalExecutable;
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/ |
| H A D | raw_api_test.cc | 373 std::vector<std::unique_ptr<xla::LocalExecutable>> local_executables = in XlaCompiledProgramShape()
376 std::unique_ptr<xla::LocalExecutable> local_executable = in XlaCompiledProgramShape()
|