Home
last modified time | relevance | path

Searched defs:report_instruction_count (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h44 int64 report_instruction_count = 5; member
165 explicit SpmdLogger(int64_t report_instruction_count) in SpmdLogger()
Dspmd_partitioner.cc97 const HloModule& module, int64_t report_instruction_count) { in ReportBeforePartition()
118 const HloModule& module, int64_t report_instruction_count) { in ReportAfterPartition()
132 int64_t report_instruction_count) { in ReportMemoryUsage()