Searched defs:tpu_program_group (Results 1 – 12 of 12) 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
32 TpuProgramGroupInterface* tpu_program_group) { in Compile()
43 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group; member
137 TpuProgramGroupInterface* tpu_program_group) { in LookupPersistentCompilationCacheAndFillCaches()
152 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group; member
271 TpuProgramGroup* tpu_program_group = in CompileAndBuild() local307 TpuProgramGroup* tpu_program_group = in CompileAndBuild() local
596 TpuProgramGroupInterface* tpu_program_group) { in CompileLocallyAndFillHostCache()695 [&](TpuProgramGroupInterface* tpu_program_group) { in ComputeInternal()
648 const tpu::TpuProgramGroup* tpu_program_group = in DoWork() local
125 const tpu::TpuProgramGroup* tpu_program_group, xla::Backend* const backend, in GetExecutableAliasConfig()191 const uint32 rng_seed, const tpu::TpuProgramGroup* tpu_program_group, in RunExecutable()233 const uint32 rng_seed, const tpu::TpuProgramGroup* tpu_program_group, in ExecuteTPUProgram()313 const tpu::TpuProgramGroup* tpu_program_group = in DoWork() local436 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()