OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:ParallelTaskAssigner
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.h
52
//
ParallelTaskAssigner
computes target parallel task counts for all HLOs
60
class
ParallelTaskAssigner
: public HloPassInterface {
65
ParallelTaskAssigner
(const int64 max_parallelism,
68
~
ParallelTaskAssigner
() override {}
parallel_task_assignment.cc
153
StatusOr<bool>
ParallelTaskAssigner
::Run(HloModule* module) {
154
XLA_VLOG_LINES(2, "
ParallelTaskAssigner
ENTRY");
164
XLA_VLOG_LINES(2, "
ParallelTaskAssigner
EXIT");
169
bool
ParallelTaskAssigner
::AssignParallelTasks(
175
bool
ParallelTaskAssigner
::AssignParallelTasksHelper(
230
void
ParallelTaskAssigner
::ComputeTargetParallelTasks(
cpu_compiler.cc
313
// Run
ParallelTaskAssigner
to assign parallel tasks to HLOs in module.
318
pipeline.AddPass<
ParallelTaskAssigner
>(max_parallelism,
[
all
...]
Completed in 1333 milliseconds