| 1312 | } |
| 1313 | |
| 1314 | void Tensor::FillDescription(TensorDescription* description) const { |
| 1315 | description->set_dtype(dtype()); |
| 1316 | shape().AsProto(description->mutable_shape()); |
| 1317 | if (buf_ != nullptr && buf_->data() != nullptr) { |
| 1318 | buf_->FillAllocationDescription( |
| 1319 | description->mutable_allocation_description()); |
| 1320 | } |
| 1321 | } |
| 1322 | |
| 1323 | gtl::InlinedVector<int64, 4> Tensor::ComputeFlatInnerDims( |
| 1324 | gtl::ArraySlice<int64> orig, int64 num_out_dims) { |
no test coverage detected