Searched refs:tpu_programs_ (Results 1 – 2 of 2) sorted by relevance
91 std::vector<bool> may_modify_variables_array(tpu_programs_.size(), false); in Initialize()92 std::vector<TPUExecutableInfoProto> executable_infos(tpu_programs_.size()); in Initialize()94 tpu_programs_.size()); in Initialize()95 std::vector<xla::HloProto> hlo_metadatas(tpu_programs_.size()); in Initialize()96 for (size_t i = 0; i < tpu_programs_.size(); ++i) { in Initialize()97 const XLA_TpuProgram* xla_tpu_program = tpu_programs_[i]; in Initialize()115 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in has_sharding_program()123 size_t TpuProgramGroup::program_count() const { return tpu_programs_.size(); } in program_count()127 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in program_size()135 for (const XLA_TpuProgram* tpu_program : tpu_programs_) { in LogProgramMemorySummary()[all …]
172 std::vector<XLA_TpuProgram*> tpu_programs_; // Not owned. variable