Move VariantTensorData::add_tensor template definition from .cc to .h file.
PiperOrigin-RevId: 343130947 Change-Id: Ied69bfa77276b3319b97328cb23f22e873851606
This commit is contained in:
parent
44e771aa50
commit
65d96cf2d5
@ -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_);
|
||||
|
@ -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_
|
||||
|
Loading…
x
Reference in New Issue
Block a user