Home
last modified time | relevance | path

Searched refs:GpuHloSchedule (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule.cc189 GpuHloSchedule::GpuHloSchedule() {} in GpuHloSchedule() function in xla::gpu::GpuHloSchedule
192 StatusOr<std::unique_ptr<GpuHloSchedule>> GpuHloSchedule::Build( in Build()
195 std::unique_ptr<GpuHloSchedule> schedule(new GpuHloSchedule); in Build()
Dgpu_hlo_schedule.h39 class GpuHloSchedule {
43 static StatusOr<std::unique_ptr<GpuHloSchedule>> Build(
61 GpuHloSchedule();
Dgpu_hlo_schedule_test.cc41 static std::unique_ptr<GpuHloSchedule> BuildGpuHloSchedule( in BuildGpuHloSchedule()
43 return GpuHloSchedule::Build(module, streams, /*pointer_size=*/8) in BuildGpuHloSchedule()
Dnvptx_compiler.cc661 std::unique_ptr<GpuHloSchedule> hlo_schedule, in RunBackend()
662 GpuHloSchedule::Build(*module, *stream_assignment, pointer_size_)); in RunBackend()