Searched defs:tpu_program (Results 1 – 2 of 2) sorted by relevance
118 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in has_sharding_program() local130 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in program_size() local138 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in LogProgramMemorySummary() local218 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in set_fingerprints() local231 const XLA_TpuProgram* TpuProgramGroup::tpu_program(int index) const { in tpu_program() function in tensorflow::tpu::TpuProgramGroup
426 const XLA_TpuProgram* tpu_program) { in TPUExecute()