Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk_schedule.cc29 void ThunkSchedule::AddDependenciesOnTransitiveOperands( in AddDependenciesOnTransitiveOperands()
50 ThunkSchedule::ThunkSchedule( in ThunkSchedule() function in xla::gpu::ThunkSchedule
84 ThunkSchedule::ThunkSchedule(std::unique_ptr<ThunkSequence> thunks) in ThunkSchedule() function in xla::gpu::ThunkSchedule
91 void ThunkSchedule::RemoveRedundantDependencyEdges() { in RemoveRedundantDependencyEdges()
146 const std::list<const Thunk*>& ThunkSchedule::DependsOn( in DependsOn()
155 string ThunkSchedule::ToString() const { in ToString()
Dthunk_schedule.h47 class ThunkSchedule {
53 ThunkSchedule(
59 explicit ThunkSchedule(std::unique_ptr<ThunkSequence> thunks);
Dgpu_executable.h74 std::unique_ptr<const ThunkSchedule> thunk_schedule;
194 const std::unique_ptr<const ThunkSchedule> thunk_schedule_;
Dgpu_compiler.cc557 std::unique_ptr<ThunkSchedule>* thunk_schedule, in CompileModuleToLlvmIrImpl()
653 *thunk_schedule = absl::make_unique<ThunkSchedule>( in CompileModuleToLlvmIrImpl()
901 std::unique_ptr<ThunkSchedule> thunk_schedule; in RunBackend()
991 std::unique_ptr<ThunkSchedule> thunk_schedule; in CompileModuleToLlvmIr()
1134 auto thunk_schedule = absl::make_unique<ThunkSchedule>( in CompileLmhloToExecutable()