Move VariantTensorData::add_tensor template definition from .cc to .h file.

PiperOrigin-RevId: 343130947
Change-Id: Ied69bfa77276b3319b97328cb23f22e873851606
This commit is contained in:
Edward Loper 2020-11-18 12:19:26 -08:00 committed by TensorFlower Gardener
parent 44e771aa50
commit 65d96cf2d5
2 changed files with 6 additions and 6 deletions

View File

@ -39,12 +39,6 @@ Tensor* VariantTensorData::add_tensors() {
return &(tensors_[tensors_.size() - 1]);
}
template <typename... TensorConstructorArgs>
Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) {
tensors_.emplace_back(std::forward<TensorConstructorArgs>(args)...);
return &tensors_.back();
}
void VariantTensorData::ToProto(VariantTensorDataProto* proto) const {
proto->set_type_name(type_name());
proto->set_metadata(metadata_);

View File

@ -125,6 +125,12 @@ class VariantTensorData {
// For backwards compatibility for when this was a proto
std::string ProtoDebugString(const VariantTensorData& object);
template <typename... TensorConstructorArgs>
Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) {
tensors_.emplace_back(std::forward<TensorConstructorArgs>(args)...);
return &tensors_.back();
}
} // namespace tensorflow
#endif // TENSORFLOW_CORE_FRAMEWORK_VARIANT_TENSOR_DATA_H_