Searched defs:xla_tpu_program (Results 1 – 1 of 1) sorted by relevance
37 const XLA_TpuProgram* xla_tpu_program) { in ConstructExecutableInfo()53 const XLA_TpuProgram* xla_tpu_program) { in ConstructHostTransferInfo()69 const XLA_TpuProgram* xla_tpu_program) { in ConstructHloMetadata()97 const XLA_TpuProgram* xla_tpu_program = tpu_programs_[i]; in Initialize() local336 auto* xla_tpu_program = OpsApiFn()->TpuProgram_NewFn(); in DeserializeFromRpcResponseProtos() local