Home
last modified time | relevance | path

Searched refs:cpu_executable_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.h485 return cpu_executable_->shared_module()->name(); in name()
493 return cpu_executable_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
512 cpu_executable_->shared_module()}; in GetHloModules()
567 std::shared_ptr<Executable> cpu_executable_; variable
H A Dtfrt_cpu_pjrt_client.cc1171 cpu_executable_(std::move(cpu_executable)), in TfrtCpuExecutable()
1186 cpu_executable_->module().entry_computation_layout(); in TfrtCpuExecutable()
1221 *cpu_executable_->shared_module(), tuple_inputs)); in SetUpDonation()
1432 tensorflow::down_cast<cpu::CpuExecutable*>(cpu_executable_.get()); in ExecuteHelper()
1522 cpu_executable_copy = cpu_executable_, in ExecuteHelper()
1569 const Shape& result_shape = cpu_executable_->result_shape(); in ExecuteHelper()