Searched refs:program_count (Results 1 – 7 of 7) 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()
45 virtual size_t program_count() const = 0;
427 if (tpu_program_group->program_count() != 1) { in RegisterXLAFingerprints()428 LOG(INFO) << "Found " << tpu_program_group->program_count() in RegisterXLAFingerprints()
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
46 for (int i = 0; i < tpu_program_group.program_count(); ++i) { in PopulateEntry()
113 size_t program_count() const override;
466 bool compilation_success = entry->tpu_program_group->program_count() > 0; in CompileIfKeyAbsentHelper()574 proto_index >= cache_entry->tpu_program_group->program_count()) { in Lookup()