Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc60 int num_slots = cm.slot_bytes_[local_id].size(); in MergeFromLocal()
65 if (slot_bytes_[global_id].empty()) { in MergeFromLocal()
66 slot_bytes_[global_id].resize(num_slots); in MergeFromLocal()
68 CHECK_EQ(num_slots, slot_bytes_[global_id].size()); in MergeFromLocal()
71 slot_bytes_[global_id][s] += cm.slot_bytes_[local_id][s]; in MergeFromLocal()
84 int num_slots = cm.slot_bytes_[i].size(); in MergeFromGlobal()
87 if (slot_bytes_[i].empty()) { in MergeFromGlobal()
88 slot_bytes_[i].resize(num_slots); in MergeFromGlobal()
90 CHECK_EQ(num_slots, slot_bytes_[i].size()); in MergeFromGlobal()
93 slot_bytes_[i][s] += cm.slot_bytes_[i][s]; in MergeFromGlobal()
[all …]
Dcostmodel.h201 std::vector<gtl::InlinedVector<Bytes, 2>> slot_bytes_; variable