HomeSort by relevance Sort by last modified time
    Searched refs:GpuHloSchedule (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_hlo_schedule.h 39 class GpuHloSchedule {
41 // Constructs an GpuHloSchedule for the given module, based on the given
43 static StatusOr<std::unique_ptr<GpuHloSchedule>> Build(
61 GpuHloSchedule();
gpu_hlo_schedule.cc 189 GpuHloSchedule::GpuHloSchedule() {}
192 StatusOr<std::unique_ptr<GpuHloSchedule>> GpuHloSchedule::Build(
195 std::unique_ptr<GpuHloSchedule> schedule(new GpuHloSchedule);
gpu_hlo_schedule_test.cc 41 static std::unique_ptr<GpuHloSchedule> BuildGpuHloSchedule(
43 return GpuHloSchedule::Build(module, streams, /*pointer_size=*/8)
nvptx_compiler.cc 661 std::unique_ptr<GpuHloSchedule> hlo_schedule,
662 GpuHloSchedule::Build(*module, *stream_assignment, pointer_size_));
    [all...]

Completed in 110 milliseconds