Searched defs:tpu_program (Results 1 – 2 of 2) sorted by relevance
115 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in has_sharding_program() local127 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in program_size() local135 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in LogProgramMemorySummary() local217 const XLA_TpuProgram* TpuProgramGroup::tpu_program(int index) const { in tpu_program() function in tensorflow::tpu::TpuProgramGroup
422 const XLA_TpuProgram* tpu_program) { in TPUExecute()