Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc380 computation_nest_level_( in CostAnalysisPrefetchIntervalPicker()
410 computation_nest_level_[logical_time] = computation_nest_level; in CostAnalysisPrefetchIntervalPicker()
489 int end_nest_level = computation_nest_level_[end_time]; in LatestPrefetchStartTime()
496 (computation_nest_level_[latest_prefetch_time] != end_nest_level || in LatestPrefetchStartTime()
517 int end_nest_level = computation_nest_level_[prefetch_end_time]; in PreferredPrefetchStartTime()
523 if (computation_nest_level_[prefetch_start_time] == end_nest_level && in PreferredPrefetchStartTime()
538 computation_nest_level_[original_prefetch_end_time]; in LatestPrefetchEndTime()
541 computation_nest_level_[new_prefetch_end_time] != original_nest_level; in LatestPrefetchEndTime()
562 int end_nest_level = computation_nest_level_[end_logical_time_]; in Begin()
574 (computation_nest_level_[earliest_prefetch_time_] != end_nest_level || in Begin()
[all …]
Dmemory_space_assignment.h359 std::vector<int> computation_nest_level_; variable