| 49 | }; |
| 50 | |
| 51 | Stack(const DataType& elem_type, const string& stack_name, int max_size) |
| 52 | : elem_type_(elem_type), |
| 53 | stack_name_(stack_name), |
| 54 | max_size_(max_size), |
| 55 | closed_(false) {} |
| 56 | |
| 57 | Status Push(const TensorAndAllocation& value) { |
| 58 | mutex_lock l(mu_); |