Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc163 somas_json[kContiguousSize] = contiguous_tensors_list_.size(); in SaveSomasResult()
257 if (contiguous_size != contiguous_tensors_list_.size()) { in VerifySomasResult()
258 …NG) << "Mismatch contiguous size " << contiguous_size << " vs " << contiguous_tensors_list_.size(); in VerifySomasResult()
359 << contiguous_tensors_list_.size() << " contiguous lists"; in InitSomasTensors()
936 contiguous_tensors_list_.push_back(inputs); in GenContiguousList()
954 contiguous_tensors_list_.push_back(outputs); in GenContiguousList()
1165 vector<vector<size_t>> contiguous_tensors_list_removed = contiguous_tensors_list_; in Assign()
1168 contiguous_tensors_list_to_remove.insert(contiguous_tensors_list_[ref_list_pair.second]); in Assign()
1172 for (auto contiguous_list : contiguous_tensors_list_) { in Assign()
1264 …for (size_t index = 0; index < contiguous_tensors_list_.size() && (!found_first || !found_second);… in GetContiguousListContainRefTensor()
[all …]
Dsomas.h85 std::vector<vector<size_t>> contiguous_tensors_list_; variable