Searched refs:program_count (Results 1 – 7 of 7) sorted by relevance
87 CHECK_GE(tpu_program_group->program_count(), 0); in SerializeCacheEntryToBufferSlices()89 CHECK_LT(cache_entry.core_index(), tpu_program_group->program_count()); in SerializeCacheEntryToBufferSlices()
87 CHECK_EQ(program_count(), 0) << "Reinitialization of an existing " in Initialize()126 size_t TpuProgramGroup::program_count() const { return tpu_programs_.size(); } in program_count() function in tensorflow::tpu::TpuProgramGroup
471 if (tpu_program_group->program_count() != 1) { in RegisterXLAFingerprints()472 LOG(INFO) << "Found " << tpu_program_group->program_count() in RegisterXLAFingerprints()
45 virtual size_t program_count() const = 0;
468 bool compilation_success = entry->tpu_program_group->program_count() > 0; in CompileIfKeyAbsentHelper()576 proto_index >= cache_entry->tpu_program_group->program_count()) { in Lookup()
46 for (int i = 0; i < tpu_program_group.program_count(); ++i) { in PopulateEntry()
113 size_t program_count() const override;