Searched refs:program_count (Results 1 – 6 of 6) sorted by relevance
86 CHECK_GE(tpu_program_group->program_count(), 0); in SerializeCacheEntryToBufferSlices()88 CHECK_LT(cache_entry.core_index(), tpu_program_group->program_count()); in SerializeCacheEntryToBufferSlices()
43 virtual size_t program_count() const = 0;
46 for (int i = 0; i < tpu_program_group.program_count(); ++i) { in PopulateEntry()
87 CHECK_EQ(program_count(), 0) << "Reinitialization of an existing " in Initialize()123 size_t TpuProgramGroup::program_count() const { return tpu_programs_.size(); } in program_count() function in tensorflow::tpu::TpuProgramGroup
113 size_t program_count() const override;
464 bool compilation_success = entry->tpu_program_group->program_count() > 0; in CompileIfKeyAbsentHelper()572 proto_index >= cache_entry->tpu_program_group->program_count()) { in Lookup()