Home
last modified time | relevance | path

Searched refs:executable_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.cc30 executable_(std::move(executable)), in PyExecutable()
43 VLOG(1) << "Fingerprint for executable " << executable_->name() << ": " in PyExecutable()
63 devices.reserve(executable_->addressable_devices().size()); in AddressableDevices()
64 for (PjRtDevice* device : executable_->addressable_devices()) { in AddressableDevices()
78 TF_ASSIGN_OR_RETURN(output_buffers, executable_->Execute({args}, options_)); in PjRtExecute()
99 executable_->Execute({arg_buffers}, options_)); in Execute()
124 executable_->Execute(arg_buffers, options_)); in ExecuteOnLocalDevices()
143 int num_computations = executable_->addressable_devices().size(); in ExecuteShardedOnLocalDevices()
168 executable_->Execute(arg_buffers, options_)); in ExecuteShardedOnLocalDevices()
187 return executable_->GetHloModules(); in HloModules()
Dpy_executable.h48 return executable_->addressable_device_logical_ids(); in addressable_device_logical_ids()
54 return executable_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
57 void Delete() { return executable_->Delete(); } in Delete()
80 const PjRtExecutable& pjrt_executable() const { return *executable_; } in pjrt_executable()
86 std::unique_ptr<PjRtExecutable> executable_; variable
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc47 : executable_(std::move(executable)), in LocalExecutable()
122 executable_->module_config().entry_computation_layout(); in RunHelper()
135 executable_.get(), i, in RunHelper()
253 if (executable_->dumping_snapshot()) { in RunAsync()
254 snapshot = DumpArguments(backend_, executable_.get(), arguments, stream); in RunAsync()
258 executable_->ExecuteAsyncOnStreamWrapper( in RunAsync()
294 if (executable_->dumping_snapshot()) { in RunAsync()
306 DumpArguments(backend_, executable_.get(), shaped_buffer_ptrs, stream); in RunAsync()
310 executable_->ExecuteAsyncOnStreamWrapper( in RunAsync()
Dlocal_client.h74 Executable* executable() const { return executable_.get(); } in executable()
118 std::unique_ptr<Executable> executable_; variable
/external/protobuf/conformance/
Dconformance_test_runner.cc294 std::unique_ptr<char[]> executable(new char[executable_.size() + 1]); in SpawnTestProgram()
295 memcpy(executable.get(), executable_.c_str(), executable_.size()); in SpawnTestProgram()
296 executable[executable_.size()] = '\0'; in SpawnTestProgram()
Dconformance_test.h92 : child_pid_(-1), executable_(executable) {} in ForkPipeRunner()
110 std::string executable_; variable
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.h65 std::unique_ptr<xla::LocalExecutable> executable_; variable
Dxla_jit_compiled_cpu_function.cc142 jit->executable_ = std::move(executable); in Compile()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc82 executable_(executable), in XlaExecutableClosure()
91 xla::LocalExecutable* executable() const { return executable_; } in executable()
102 xla::LocalExecutable* executable_; member in tensorflow::__anon46563ba90111::XlaExecutableClosure