| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_compilation_cache.h | 32 explicit XRTCompilationCacheEntry(xla::LocalExecutable* executable) in XRTCompilationCacheEntry()
36 xla::LocalExecutable* get_executable() const { return executable; } in get_executable()
39 xla::LocalExecutable* executable;
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>*)>&
|
| D | xrt_compilation_cache.cc | 161 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>& in InitializeEntry()
182 std::unique_ptr<xla::LocalExecutable> program; in InitializeEntry()
207 const std::function<Status(std::unique_ptr<xla::LocalExecutable>*)>& in CompileIfKeyAbsent()
|
| /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()
143 StatusOr<ScopedShapedBuffer> LocalExecutable::Run( in Run()
178 StatusOr<ScopedShapedBuffer> LocalExecutable::ExecuteAndDump( in ExecuteAndDump()
193 Status LocalExecutable::RecordArguments( in RecordArguments()
204 Status LocalExecutable::RecordResult(const ShapedBuffer* result, in RecordResult()
212 StatusOr<Literal> LocalExecutable::LiteralFromShapedBuffer( in LiteralFromShapedBuffer()
244 StatusOr<std::unique_ptr<LocalExecutable>> LocalClient::Compile( in Compile()
257 return absl::WrapUnique(new LocalExecutable(std::move(executable), in Compile()
|
| D | local_client.h | 38 class LocalExecutable {
57 LocalExecutable(std::unique_ptr<Executable> executable, Backend* backend,
119 StatusOr<std::unique_ptr<LocalExecutable>> Compile(
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.h | 77 xla::LocalExecutable** out_executable);
86 xla::LocalExecutable** out_executable);
130 xla::LocalExecutable** out_executable);
136 std::unique_ptr<xla::LocalExecutable>* executable);
159 std::unique_ptr<xla::LocalExecutable> executable GUARDED_BY(mu);
|
| D | xla_compile_on_demand_op.h | 47 xla::LocalExecutable** executable);
50 xla::LocalExecutable* executable);
|
| D | xla_compile_on_demand_op.cc | 53 xla::LocalExecutable* executable) { in Run()
116 xla::LocalExecutable** executable) { in Compile()
213 xla::LocalExecutable* executable; in Compute()
|
| D | xla_compilation_cache.cc | 143 std::unique_ptr<xla::LocalExecutable>* executable) { in BuildExecutable()
173 xla::LocalExecutable** out_executable) { in Compile()
203 xla::LocalExecutable** out_executable) { in CompileSingleOp()
233 xla::LocalExecutable** out_executable) { in CompileImpl()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 66 std::unique_ptr<xla::LocalExecutable>* program);
84 std::unique_ptr<xla::LocalExecutable>* program) { in Compile()
175 key, &uid, [&](std::unique_ptr<xla::LocalExecutable>* program) { in Compute()
186 xla::LocalExecutable* executable = entry->get().get_executable(); in Compute()
|
| D | xrt_execute_op.cc | 216 xla::LocalExecutable* executable = entry->get().get_executable(); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.h | 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);
|
| D | 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;
|
| D | local_computation_builder.cc | 210 LocalExecutable::LocalExecutable( in LocalExecutable() function in xla::swig::LocalExecutable
211 std::unique_ptr<xla::LocalExecutable> executable, in LocalExecutable()
217 std::vector<int> LocalExecutable::DeviceOrdinals() const { in DeviceOrdinals()
227 StatusOr<LocalShapedBuffer*> LocalExecutable::Execute( in Execute()
264 StatusOr<LocalShapedBufferTuple*> LocalExecutable::ExecutePerReplica( in ExecutePerReplica()
339 StatusOr<LocalExecutable*> Computation::Compile( in Compile()
360 return new LocalExecutable(std::move(local_executable), in Compile()
867 void DeleteLocalExecutable(LocalExecutable* computation) { delete computation; } in DeleteLocalExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_metadata_test.cc | 49 std::unique_ptr<LocalExecutable> executable, in TEST_F()
75 std::unique_ptr<LocalExecutable> executable = in TEST_F()
|
| D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
68 std::unique_ptr<LocalExecutable> executable, in TestWithDeviceCount()
|
| D | local_client_test_base.cc | 191 std::unique_ptr<LocalExecutable> executable, in ExecuteLocally()
|
| D | xla_hlo_profile_test.cc | 161 std::unique_ptr<LocalExecutable> local_executable, in ExecuteAndFetchProfile()
|
| D | local_client_execute_test.cc | 764 std::unique_ptr<LocalExecutable> executable = in XLA_TEST_F()
932 std::unique_ptr<LocalExecutable> executable = in BM_LocalClientOverhead()
|
| D | dynamic_ops_test.cc | 782 std::unique_ptr<LocalExecutable> executable = in BM_DynamicSlice()
|
| D | fusion_test.cc | 885 std::unique_ptr<LocalExecutable> executable = in BM_ParallelFusion()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 114 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
128 xla::LocalExecutable* executable() const { return executable_; } in executable()
139 xla::LocalExecutable* executable_;
260 xla::LocalExecutable** executable) { in CompileToLocalExecutable()
329 xla::LocalExecutable* executable; in Compute()
446 xla::LocalExecutable* executable; in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 93 StatusOr<std::unique_ptr<LocalExecutable>> CompileExecutable( in CompileExecutable()
197 LocalExecutable* executable, in ReplayComputation()
363 std::vector<StatusOr<std::unique_ptr<LocalExecutable>>> executables; in RealMain()
385 LocalExecutable* executable = executables[i].ValueOrDie().get(); in RealMain()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.h | 64 std::unique_ptr<xla::LocalExecutable> executable_;
|
| D | xla_jit_compiled_cpu_function.cc | 113 TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::LocalExecutable> executable, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.h | 205 friend class LocalExecutable;
|