Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc187 simple_tensor_mask_.reserve(dtypes_.size()); in Initialize()
190 simple_tensor_mask_.push_back(true); in Initialize()
193 simple_tensor_mask_.push_back(false); in Initialize()
232 if (simple_tensor_mask_[i]) { in WriteTensors()
252 if (simple_tensor_mask_[i]) { in WriteTensors()
687 simple_tensor_mask_.reserve(dtypes_.size()); in Initialize()
690 simple_tensor_mask_.push_back(true); in Initialize()
693 simple_tensor_mask_.push_back(false); in Initialize()
733 for (int i = 0, end = simple_tensor_mask_.size(); i < end; ++i) { in ReadTensors()
734 if (simple_tensor_mask_[i]) { in ReadTensors()
[all …]
Dsnapshot_util.h161 std::vector<bool> simple_tensor_mask_; // true for simple, false for complex. variable
280 std::vector<bool> simple_tensor_mask_; // true for simple, false for complex. variable