| /external/tensorflow/tensorflow/compiler/xrt/ |
| xrt_compilation_cache.h | 32 explicit XRTCompilationCacheEntry(xla::LocalExecutable* executable)
36 xla::LocalExecutable* get_executable() const { return executable; }
39 xla::LocalExecutable* executable;
107 // the xla::LocalExecutable into its passed argument. It should return OK
112 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
149 std::unique_ptr<xla::LocalExecutable> program;
206 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
|
| xrt_compilation_cache.cc | 161 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
182 std::unique_ptr<xla::LocalExecutable> program;
207 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>&
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compile_on_demand_op.h | 47 xla::LocalExecutable** executable);
50 xla::LocalExecutable* executable);
|
| xla_compilation_cache.h | 68 // xla::LocalExecutable and sets `executable` to point to it. The resulting
77 xla::LocalExecutable** out_executable);
86 xla::LocalExecutable** out_executable);
130 xla::LocalExecutable** out_executable);
133 // XLA computation already, and generates an XLA LocalExecutable `executable`.
136 std::unique_ptr<xla::LocalExecutable>* executable);
159 std::unique_ptr<xla::LocalExecutable> executable GUARDED_BY(mu);
|
| xla_compile_on_demand_op.cc | 53 xla::LocalExecutable* executable) {
116 xla::LocalExecutable** executable) {
213 xla::LocalExecutable* executable;
|
| xla_compilation_cache.cc | 143 std::unique_ptr<xla::LocalExecutable>* executable) {
173 xla::LocalExecutable** out_executable) {
203 xla::LocalExecutable** out_executable) {
233 xla::LocalExecutable** out_executable) {
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| local_client.h | 38 class LocalExecutable {
57 LocalExecutable(std::unique_ptr<Executable> executable, Backend* backend,
114 // Build and return a LocalExecutable object. The executable is compiled using
119 StatusOr<std::unique_ptr<LocalExecutable>> Compile(
|
| local_client.cc | 44 LocalExecutable::LocalExecutable(std::unique_ptr<Executable> executable,
54 Status LocalExecutable::ValidateExecutionOptions(
143 StatusOr<ScopedShapedBuffer> LocalExecutable::Run(
178 StatusOr<ScopedShapedBuffer> LocalExecutable::ExecuteAndDump(
193 Status LocalExecutable::RecordArguments(
204 Status LocalExecutable::RecordResult(const ShapedBuffer* result,
212 StatusOr<Literal> LocalExecutable::LiteralFromShapedBuffer(
244 StatusOr<std::unique_ptr<LocalExecutable>> LocalClient::Compile(
257 return absl::WrapUnique(new LocalExecutable(std::move(executable)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| hlo_metadata_test.cc | 49 std::unique_ptr<LocalExecutable> executable,
75 std::unique_ptr<LocalExecutable> executable =
|
| multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client,
68 std::unique_ptr<LocalExecutable> executable,
|
| local_client_test_base.cc | 191 std::unique_ptr<LocalExecutable> executable,
|
| xla_hlo_profile_test.cc | 161 std::unique_ptr<LocalExecutable> local_executable,
|
| local_client_execute_test.cc | [all...] |
| dynamic_ops_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_jit_compiled_cpu_function.h | 64 std::unique_ptr<xla::LocalExecutable> executable_;
|
| xla_jit_compiled_cpu_function.cc | 113 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::LocalExecutable> executable,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| xrt_compile_ops.cc | 66 std::unique_ptr<xla::LocalExecutable>* program);
84 std::unique_ptr<xla::LocalExecutable>* program) {
175 key, &uid, [&](std::unique_ptr<xla::LocalExecutable>* program) {
186 xla::LocalExecutable* executable = entry->get().get_executable();
|
| xrt_execute_op.cc | 216 xla::LocalExecutable* executable = entry->get().get_executable();
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.h | 123 // device-allocated literals. Specifically, wraps an XLA LocalExecutable.
124 class LocalExecutable {
126 LocalExecutable(std::unique_ptr<xla::LocalExecutable> executable,
147 const std::unique_ptr<xla::LocalExecutable> executable_;
160 StatusOr<LocalExecutable*> Compile(
471 void DeleteLocalExecutable(LocalExecutable* computation);
|
| local_computation_builder.i | 88 %typemap(out) StatusOr<xla::swig::LocalExecutable*> {
93 $typemap(out, xla::swig::LocalExecutable*)
272 %unignore xla::swig::LocalExecutable;
273 %unignore xla::swig::LocalExecutable::DeviceOrdinals;
274 %unignore xla::swig::LocalExecutable::Execute;
275 %unignore xla::swig::LocalExecutable::ExecutePerReplica;
|
| local_computation_builder.cc | 210 LocalExecutable::LocalExecutable(
211 std::unique_ptr<xla::LocalExecutable> executable,
217 std::vector<int> LocalExecutable::DeviceOrdinals() const {
227 StatusOr<LocalShapedBuffer*> LocalExecutable::Execute(
264 StatusOr<LocalShapedBufferTuple*> LocalExecutable::ExecutePerReplica(
339 StatusOr<LocalExecutable*> Computation::Compile(
360 return new LocalExecutable(std::move(local_executable),
867 void DeleteLocalExecutable(LocalExecutable* computation) { delete computation; }
|
| xla_client.py | 904 LocalExecutable, which in turn can be locally executed.
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| xla_ops.cc | 114 xla::LocalClient* client, xla::LocalExecutable* executable,
128 xla::LocalExecutable* executable() const { return executable_; }
139 xla::LocalExecutable* executable_;
260 xla::LocalExecutable** executable) {
329 xla::LocalExecutable* executable;
446 xla::LocalExecutable* executable;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| replay_computation.cc | 93 StatusOr<std::unique_ptr<LocalExecutable>> CompileExecutable(
197 LocalExecutable* executable,
363 std::vector<StatusOr<std::unique_ptr<LocalExecutable>>> executables;
385 LocalExecutable* executable = executables[i].ValueOrDie().get();
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| service.h | 205 friend class LocalExecutable;
|
