|
|
@ -47,13 +47,6 @@ class TensorArray {
|
|
|
|
// max number of values allowed to store.
|
|
|
|
// max number of values allowed to store.
|
|
|
|
const size_t MAX_SIZE{100000};
|
|
|
|
const size_t MAX_SIZE{100000};
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
|
|
|
* Inputs:
|
|
|
|
|
|
|
|
* - value_shared: share memory between tensors.
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
explicit TensorArray(bool values_shared = true)
|
|
|
|
|
|
|
|
: values_shared_(values_shared) {}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
/*
|
|
|
|
* Read the value at location `index` in the `TensorArray`.
|
|
|
|
* Read the value at location `index` in the `TensorArray`.
|
|
|
|
*/
|
|
|
|
*/
|
|
|
@ -111,7 +104,6 @@ class TensorArray {
|
|
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
private:
|
|
|
|
mutable std::vector<LoDTensor> values_;
|
|
|
|
mutable std::vector<LoDTensor> values_;
|
|
|
|
bool values_shared_;
|
|
|
|
|
|
|
|
}; // class TensorArray
|
|
|
|
}; // class TensorArray
|
|
|
|
|
|
|
|
|
|
|
|
} // namespace framework
|
|
|
|
} // namespace framework
|
|
|
|