Searched defs:tpu_program_group (Results 1 – 13 of 13) sorted by relevance
31 const TpuProgramGroupInterface* tpu_program_group, int core_index) in TpuCompilationCacheEntry()37 const TpuProgramGroupInterface* tpu_program_group() const { in tpu_program_group() function
44 TpuProgramGroup tpu_program_group) { in PopulateEntry()96 TpuProgramGroup tpu_program_group; in InitializeEntry() local
54 auto tpu_program_group = absl::make_unique<TpuProgramGroup>(); in DeserializeRpcResponseToCacheEntry() local82 const TpuProgramGroup* tpu_program_group = in SerializeCacheEntryToBufferSlices() local
150 TpuProgramGroupInterface* tpu_program_group) { in CompileLocallyAndFillHostCache()253 [&](TpuProgramGroupInterface* tpu_program_group) { in ComputeInternal()425 TpuProgramGroupInterface* tpu_program_group, uint64 fingerprint) { in RegisterXLAFingerprints()
32 TpuProgramGroupInterface* tpu_program_group) { in Compile()
43 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group; member
135 TpuProgramGroupInterface* tpu_program_group) { in LookupPersistentCompilationCacheAndFillCaches()
152 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group; member
292 TpuProgramGroup* tpu_program_group = in CompileAndBuild() local328 TpuProgramGroup* tpu_program_group = in CompileAndBuild() local
153 const tpu::TpuProgramGroupInterface* tpu_program_group = in DoTpuExecute() local
644 const tpu::TpuProgramGroup* tpu_program_group = in DoWork() local
125 const tpu::TpuProgramGroup* tpu_program_group, xla::Backend* const backend, in GetExecutableAliasConfig()192 const uint32 rng_seed, const tpu::TpuProgramGroup* tpu_program_group, in RunExecutable()234 const uint32 rng_seed, const tpu::TpuProgramGroup* tpu_program_group, in ExecuteTPUProgram()315 const tpu::TpuProgramGroup* tpu_program_group = in DoWork() local438 const tpu::TpuProgramGroup* tpu_program_group = in DoWork() local
85 tensorflow::tpu::TpuProgramGroupInterface* tpu_program_group) { in Compile()202 [&](tpu::TpuProgramGroupInterface* tpu_program_group) { in Compute()