| 251 | } |
| 252 | |
| 253 | Status SetElemShape(const PartialTensorShape& candidate) { |
| 254 | mutex_lock l(mu_); |
| 255 | PartialTensorShape new_element_shape_; |
| 256 | Status s = element_shape_.MergeWith(candidate, &new_element_shape_); |
| 257 | if (!s.ok()) { |
| 258 | return s; |
| 259 | } |
| 260 | element_shape_ = new_element_shape_; |
| 261 | return Status::OK(); |
| 262 | } |
| 263 | |
| 264 | string DebugString() const override { |
| 265 | mutex_lock l(mu_); |