Searched refs:GpuHloSchedule (Results 1 – 4 of 4) sorted by relevance
189 GpuHloSchedule::GpuHloSchedule() {} in GpuHloSchedule() function in xla::gpu::GpuHloSchedule192 StatusOr<std::unique_ptr<GpuHloSchedule>> GpuHloSchedule::Build( in Build()195 std::unique_ptr<GpuHloSchedule> schedule(new GpuHloSchedule); in Build()
39 class GpuHloSchedule {43 static StatusOr<std::unique_ptr<GpuHloSchedule>> Build(61 GpuHloSchedule();
41 static std::unique_ptr<GpuHloSchedule> BuildGpuHloSchedule( in BuildGpuHloSchedule()43 return GpuHloSchedule::Build(module, streams, /*pointer_size=*/8) in BuildGpuHloSchedule()
661 std::unique_ptr<GpuHloSchedule> hlo_schedule, in RunBackend()662 GpuHloSchedule::Build(*module, *stream_assignment, pointer_size_)); in RunBackend()