Searched refs:TensorUsageWithIndex (Results 1 – 3 of 3) sorted by relevance
36 struct TensorUsageWithIndex { struct40 TensorUsageWithIndex(const TensorUsageRecord<TensorSizeT>* usage_record, in TensorUsageWithIndex() argument45 bool CompareBySize(const TensorUsageWithIndex<size_t>& first,46 const TensorUsageWithIndex<size_t>& second);51 using TaskProfile = std::vector<TensorUsageWithIndex<size_t>>;
28 bool CompareBySize(const TensorUsageWithIndex<size_t>& first, in CompareBySize()29 const TensorUsageWithIndex<size_t>& second) { in CompareBySize()
76 std::vector<TensorUsageWithIndex<size_t>> ordered_records; in GreedyBySizeAssignment()