Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc62 int num_slots = cm.slot_bytes_[local_id].size(); in MergeFromLocal()
67 if (slot_bytes_[global_id].empty()) { in MergeFromLocal()
68 slot_bytes_[global_id].resize(num_slots); in MergeFromLocal()
70 CHECK_EQ(num_slots, slot_bytes_[global_id].size()); in MergeFromLocal()
73 slot_bytes_[global_id][s] += cm.slot_bytes_[local_id][s]; in MergeFromLocal()
86 int num_slots = cm.slot_bytes_[i].size(); in MergeFromGlobal()
89 if (slot_bytes_[i].empty()) { in MergeFromGlobal()
90 slot_bytes_[i].resize(num_slots); in MergeFromGlobal()
92 CHECK_EQ(num_slots, slot_bytes_[i].size()); in MergeFromGlobal()
95 slot_bytes_[i][s] += cm.slot_bytes_[i][s]; in MergeFromGlobal()
[all …]
Dcostmodel.h209 std::vector<gtl::InlinedVector<Bytes, 2>> slot_bytes_; variable