Home
last modified time | relevance | path

Searched refs:start_ordinal (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc629 int64_t start_ordinal, int64_t limit_ordinal) const { in MaxInRange() argument
630 int64_t max = Calculate(start_ordinal); in MaxInRange()
631 for (int64_t i = start_ordinal + 1; i < limit_ordinal; ++i) { in MaxInRange()
700 int64_t OffsetCalculation::MaxInRange(int64_t start_ordinal, in MaxInRange() argument
703 return Calculate(start_ordinal); in MaxInRange()
706 return std::max(Calculate(start_ordinal), Calculate(limit_ordinal - 1)); in MaxInRange()
708 int64_t max = Calculate(start_ordinal); in MaxInRange()
709 for (int64_t i = start_ordinal + 1; i < limit_ordinal; ++i) { in MaxInRange()
Dspmd_partitioner_util.h246 int64_t MaxInRange(int64_t start_ordinal, int64_t limit_ordinal) const;
290 int64_t MaxInRange(int64_t start_ordinal, int64_t limit_ordinal) const;
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.cc50 const HloComputation* comp, int64_t start_ordinal, in CalculatePostOrderScheduleHelper() argument
52 int64_t ordinal = start_ordinal; in CalculatePostOrderScheduleHelper()