| /external/tensorflow/tensorflow/compiler/xrt/ |
| 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>*)>&
151 std::unique_ptr<xla::LocalExecutable> program;
208 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.h | 78 xla::LocalExecutable** out_executable);
89 xla::LocalExecutable** out_executable);
134 xla::LocalExecutable** out_executable);
140 std::unique_ptr<xla::LocalExecutable>* executable);
163 std::unique_ptr<xla::LocalExecutable> executable TF_GUARDED_BY(mu);
|
| D | get_compiler_ir.cc | 36 static xla::StatusOr<xla::LocalExecutable*> GetLocalExecutable( in GetLocalExecutable()
42 xla::LocalExecutable* executable = nullptr; in GetLocalExecutable()
143 xla::StatusOr<xla::LocalExecutable*> executable = GetLocalExecutable( in GetCompilerIr()
154 xla::StatusOr<xla::LocalExecutable*> executable = GetLocalExecutable( in GetCompilerIr()
|
| D | xla_compile_on_demand_op.h | 50 xla::LocalExecutable** executable);
54 xla::LocalExecutable* executable,
|
| D | xla_compile_on_demand_op.cc | 47 xla::LocalExecutable* executable, in Run()
105 xla::LocalExecutable** executable) { in Compile()
167 xla::LocalExecutable* executable; in Compute()
|
| D | xla_compilation_cache.cc | 160 std::unique_ptr<xla::LocalExecutable>* executable) { in BuildExecutable()
191 xla::LocalExecutable** out_executable) { in Compile()
267 xla::LocalExecutable** out_executable) { in CompileSingleOp()
336 xla::LocalExecutable** out_executable) { in CompileImpl()
|
| D | xla_compilation_cache_test.cc | 67 xla::LocalExecutable* executable; in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 44 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()
172 StatusOr<ScopedShapedBuffer> LocalExecutable::Run( in Run()
186 StatusOr<ExecutionOutput> LocalExecutable::Run( in Run()
240 StatusOr<ScopedShapedBuffer> LocalExecutable::RunAsync( in RunAsync()
280 StatusOr<ExecutionOutput> LocalExecutable::RunAsync( in RunAsync()
322 StatusOr<ExecutionOutput> LocalExecutable::RunAsync( in RunAsync()
356 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> LocalClient::Compile( in Compile()
387 std::vector<std::unique_ptr<LocalExecutable>> local_executables; in Compile()
[all …]
|
| D | local_client.h | 41 class LocalExecutable {
45 LocalExecutable(std::unique_ptr<Executable> executable, Backend* backend,
143 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> Compile(
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 106 std::unique_ptr<xla::LocalExecutable>* program);
124 std::unique_ptr<xla::LocalExecutable>* program) { in Compile()
217 key, &uid, [&](std::unique_ptr<xla::LocalExecutable>* program) { in Compute()
228 xla::LocalExecutable* executable = entry->get().get_executable(); in Compute()
|
| D | xrt_execute_op.cc | 89 xla::LocalExecutable* executable, XRTMemoryManager::WorkingSet* working_set, in GetInputTuples()
298 xla::LocalExecutable* executable, in RunExecutable()
377 xla::LocalExecutable* executable, in ExecuteComputation()
397 xla::LocalExecutable* executable, in ExecuteComputation()
480 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
563 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| 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()
|
| D | local_client_execute_test.cc | 981 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in BM_LocalClientOverhead()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 78 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
91 xla::LocalExecutable* executable() const { return executable_; } in executable()
102 xla::LocalExecutable* executable_;
172 xla::LocalExecutable** executable) { in CompileToLocalExecutable()
226 xla::LocalExecutable* executable; in Compute()
375 xla::LocalExecutable* executable; in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 114 StatusOr<std::unique_ptr<LocalExecutable>> CompileExecutable( in CompileExecutable()
222 LocalExecutable* executable, in ReplayComputation()
452 std::vector<StatusOr<std::unique_ptr<LocalExecutable>>> executables; in RealMain()
482 LocalExecutable* executable = executables[i].ValueOrDie().get(); in RealMain()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.h | 65 std::unique_ptr<xla::LocalExecutable> executable_;
|
| D | xla_jit_compiled_cpu_function.cc | 123 std::unique_ptr<xla::LocalExecutable> executable = std::move(executables[0]); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.h | 604 std::vector<std::unique_ptr<LocalExecutable>> executables,
664 absl::Span<const std::shared_ptr<LocalExecutable>> executables() const { in executables()
711 std::vector<std::shared_ptr<LocalExecutable>> executables_;
|
| D | pjrt_stream_executor_client.cc | 1567 std::vector<std::unique_ptr<LocalExecutable>> executables, in PjRtStreamExecutorExecutable()
2182 std::vector<std::unique_ptr<LocalExecutable>> local_executables, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.h | 214 friend class LocalExecutable;
|
| /external/tensorflow/tensorflow/compiler/xrt/tests/ |
| D | raw_api_test.cc | 373 std::vector<std::unique_ptr<xla::LocalExecutable>> local_executables = in XlaCompiledProgramShape()
377 std::unique_ptr<xla::LocalExecutable> local_executable = in XlaCompiledProgramShape()
|