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.h161 int64 MaxInRange(int64 start_ordinal, int64 limit_ordinal) const;
205 int64 MaxInRange(int64 start_ordinal, int64 limit_ordinal) const;
Dspmd_partitioner_util.cc712 int64 start_ordinal, int64 limit_ordinal) const { in MaxInRange() argument
713 int64 max = Calculate(start_ordinal); in MaxInRange()
714 for (int64 i = start_ordinal + 1; i < limit_ordinal; ++i) { in MaxInRange()
783 int64 OffsetCalculation::MaxInRange(int64 start_ordinal, in MaxInRange() argument
786 return Calculate(start_ordinal); in MaxInRange()
789 return std::max(Calculate(start_ordinal), Calculate(limit_ordinal - 1)); in MaxInRange()
791 int64 max = Calculate(start_ordinal); in MaxInRange()
792 for (int64 i = start_ordinal + 1; i < limit_ordinal; ++i) { in MaxInRange()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.cc51 const HloComputation* comp, int64 start_ordinal, in CalculatePostOrderScheduleHelper() argument
53 int64 ordinal = start_ordinal; in CalculatePostOrderScheduleHelper()