Searched refs:xla_tpu_programs (Results 1 – 2 of 2) sorted by relevance
85 absl::Span<XLA_TpuProgram* const> xla_tpu_programs) { in Initialize() argument86 CHECK_GT(xla_tpu_programs.size(), 0); in Initialize()89 set_tpu_programs(xla_tpu_programs); in Initialize()256 XLA_TpuProgram** xla_tpu_programs = nullptr; in CompileAndBuild() local259 mesh_state, &xla_tpu_programs, in CompileAndBuild()274 absl::MakeConstSpan(&xla_tpu_programs[0], count)); in CompileAndBuild()275 OpsApiFn()->TpuProgram_FreeArrayFn(xla_tpu_programs); in CompileAndBuild()290 XLA_TpuProgram** xla_tpu_programs = nullptr; in CompileAndBuild() local293 mesh_state, &xla_tpu_programs, in CompileAndBuild()310 absl::MakeConstSpan(&xla_tpu_programs[0], count)); in CompileAndBuild()[all …]
105 void Initialize(absl::Span<XLA_TpuProgram* const> xla_tpu_programs);