Searched defs:tpu_program_group (Results 1 – 13 of 13) sorted by relevance
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
86 tensorflow::tpu::TpuProgramGroupInterface* tpu_program_group) { in Compile()203 [&](tpu::TpuProgramGroupInterface* tpu_program_group) { in Compute()
168 TpuProgramGroupInterface* tpu_program_group) { in CompileLocallyAndFillHostCache()183 TpuProgramGroupInterface* tpu_program_group) { in CompileLocallyAndFillHostCacheInternal()292 [&](TpuProgramGroupInterface* tpu_program_group) { in ComputeInternal()469 TpuProgramGroupInterface* tpu_program_group, uint64 fingerprint) { in RegisterXLAFingerprints()
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
55 auto tpu_program_group = absl::make_unique<TpuProgramGroup>(); in DeserializeRpcResponseToCacheEntry() local83 const TpuProgramGroup* tpu_program_group = in SerializeCacheEntryToBufferSlices() local
285 TpuProgramGroup* tpu_program_group = in CompileAndBuild() local321 TpuProgramGroup* tpu_program_group = in CompileAndBuild() local
43 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group; member
33 TpuProgramGroupInterface* tpu_program_group) { in Compile()
145 TpuProgramGroupInterface* tpu_program_group) { in LookupPersistentCompilationCacheAndFillCaches()
152 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group; member
153 const tpu::TpuProgramGroupInterface* tpu_program_group = in DoTpuExecute() local
643 const tpu::TpuProgramGroup* tpu_program_group = in DoWork() local