Home
last modified time | relevance | path

Searched refs:computation_nest_level_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.cc557 computation_nest_level_( in CostAnalysisPrefetchIntervalPicker()
592 computation_nest_level_[logical_time] = computation_nest_level;
698 int end_nest_level = computation_nest_level_[end_time]; in LatestPrefetchStartTime()
705 (computation_nest_level_[latest_prefetch_time] != end_nest_level || in LatestPrefetchStartTime()
726 int end_nest_level = computation_nest_level_[prefetch_end_time]; in PreferredPrefetchStartTime()
732 if (computation_nest_level_[prefetch_start_time] == end_nest_level && in PreferredPrefetchStartTime()
748 computation_nest_level_[original_prefetch_end_time]; in LatestPrefetchEndTime()
751 computation_nest_level_[new_prefetch_end_time] != original_nest_level; in LatestPrefetchEndTime()
787 int end_nest_level = computation_nest_level_[end_logical_time_]; in Begin()
798 (computation_nest_level_[earliest_prefetch_time_] != end_nest_level || in Begin()
[all …]
H A Dmemory_space_assignment.h453 std::vector<int> computation_nest_level_; variable