Searched refs:xla_tpu_program (Results 1 – 1 of 1) sorted by relevance
37 const XLA_TpuProgram* xla_tpu_program) { in ConstructExecutableInfo() argument42 xla_tpu_program, &serialized_executable_info, status.c_status); in ConstructExecutableInfo()53 const XLA_TpuProgram* xla_tpu_program) { in ConstructHostTransferInfo() argument58 xla_tpu_program, &serialized_host_transfer_info, status.c_status); in ConstructHostTransferInfo()69 const XLA_TpuProgram* xla_tpu_program) { in ConstructHloMetadata() argument74 xla_tpu_program, &serialized_hlo_metadata, status.c_status); in ConstructHloMetadata()97 const XLA_TpuProgram* xla_tpu_program = tpu_programs_[i]; in Initialize() local99 OpsApiFn()->TpuProgram_GetMayModifyVariablesFn(xla_tpu_program, in Initialize()102 executable_infos[i] = ConstructExecutableInfo(xla_tpu_program); in Initialize()103 host_transfer_infos[i] = ConstructHostTransferInfo(xla_tpu_program); in Initialize()[all …]