/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); 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);
|
D | xla_compile_on_demand_op.h | 50 xla::LocalExecutable** executable); 53 xla::LocalExecutable* executable);
|
D | xla_compile_on_demand_op.cc | 54 xla::LocalExecutable* executable) { in Run() 118 xla::LocalExecutable** executable) { in Compile() 213 xla::LocalExecutable* executable; in Compute()
|
D | xla_compilation_cache.cc | 150 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/ |
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() 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()
|
D | local_client.h | 41 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/ |
D | xrt_compile_ops.cc | 66 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()
|
D | xrt_execute_op.cc | 150 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/ |
D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute() 70 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in TestWithDeviceCount()
|
D | local_client_execute_test.cc | 960 std::unique_ptr<LocalExecutable> executable = std::move(executables[0]); in BM_LocalClientOverhead()
|
/external/tensorflow/tensorflow/compiler/jit/kernels/ |
D | xla_ops.cc | 107 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/ |
D | replay_computation.cc | 114 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/ |
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/python/ |
D | local_client.h | 308 PyLocalExecutable(std::vector<std::unique_ptr<LocalExecutable>> executables, 368 std::vector<std::shared_ptr<LocalExecutable>> executables_;
|
D | local_client.cc | 673 std::vector<std::unique_ptr<LocalExecutable>> executables, in PyLocalExecutable() 1097 std::vector<std::unique_ptr<LocalExecutable>> local_executables, in Compile()
|
D | xla_client.py | 152 return _xla.LocalExecutable.Compile(c_computation,
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | service.h | 214 friend class LocalExecutable;
|
/external/tensorflow/tensorflow/compiler/xrt/tests/ |
D | raw_api_test.cc | 288 std::vector<std::unique_ptr<xla::LocalExecutable>> local_executables = in XlaCompiledProgramShape() 292 std::unique_ptr<xla::LocalExecutable> local_executable = in XlaCompiledProgramShape()
|