Searched refs:host_transfer_info (Results 1 – 5 of 5) sorted by relevance
59 TPUHostTransferInfoProto host_transfer_info; in ConstructHostTransferInfo() local61 host_transfer_info = se_tpu::DeserializeProto<TPUHostTransferInfoProto>( in ConstructHostTransferInfo()65 return host_transfer_info; in ConstructHostTransferInfo()238 const TPUHostTransferInfoProto& TpuProgramGroup::host_transfer_info( in host_transfer_info() function in tensorflow::tpu::TpuProgramGroup
135 const TPUHostTransferInfoProto& host_transfer_info(int index) const;
752 const TPUHostTransferInfoProto& host_transfer_info = in DoWork() local753 tpu_program_group->host_transfer_info(core_index); in DoWork()756 TPUExecute(executable, host_transfer_info, in DoWork()
212 const TPUHostTransferInfoProto& host_transfer_info = in RunExecutable() local213 tpu_program_group->host_transfer_info(core_index); in RunExecutable()216 TPUExecute(executable, host_transfer_info, in RunExecutable()
350 const XLA_TpuProgram* tpu_program, TpuSerializedProto* host_transfer_info,