Searched defs:compute_index (Results 1 – 3 of 3) sorted by relevance
502 void ContinuousMemInfoHelper<Key>::AddContinuousMemInfo(bool is_input, size_t compute_index, size_t… in AddContinuousMemInfo()534 const auto &compute_index = continuous_mem_info->compute_index_; in CountContinuousMemUsage() local
48 void MemScheduler::AddContinuousMemInfo(bool is_input, size_t compute_index, size_t total_size, in AddContinuousMemInfo()
1621 for (size_t compute_index = 0; compute_index < graph.execution_order().size(); ++compute_index) { in AddCommunicationMemInfo() local