Searched refs:SpmdLogger (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 152 class SpmdLogger {
154 explicit SpmdLogger(int64 report_instruction_count) in SpmdLogger() function
201 SpmdLogger* logger);
229 int64* next_channel_id, SpmdLogger* logger,
405 int64* next_channel_id, SpmdLogger* logger,
544 SpmdLogger* logger_;
|
| D | spmd_partitioner.cc | 62 string SpmdLogger::MakeReport() { in MakeReport()
82 void SpmdLogger::RegisterLogEntry(HloInstruction* hlo, in RegisterLogEntry()
96 /* static */ string SpmdLogger::ReportBeforePartition( in ReportBeforePartition()
117 /* static */ string SpmdLogger::ReportAfterPartition( in ReportAfterPartition()
130 /* static */ string SpmdLogger::ReportMemoryUsage( in ReportMemoryUsage()
1354 int64* next_channel_id, SpmdLogger* logger, SpmdPartitionerOptions options, in SpmdPartitioningVisitor()
3634 int64* next_channel_id, SpmdLogger* logger) { in PartitionComputation()
3644 int64* next_channel_id, SpmdLogger* logger, in CreateVisitor()
3654 XLA_VLOG_LINES(1, SpmdLogger::ReportBeforePartition( in Run()
3672 SpmdLogger logger(options_.report_instruction_count); in Run()
[all …]
|