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]);
|
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 {
|
void VariantTensorData::ToProto(VariantTensorDataProto* proto) const {
|
||||||
proto->set_type_name(type_name());
|
proto->set_type_name(type_name());
|
||||||
proto->set_metadata(metadata_);
|
proto->set_metadata(metadata_);
|
||||||
|
@ -125,6 +125,12 @@ class VariantTensorData {
|
|||||||
// For backwards compatibility for when this was a proto
|
// For backwards compatibility for when this was a proto
|
||||||
std::string ProtoDebugString(const VariantTensorData& object);
|
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
|
} // namespace tensorflow
|
||||||
|
|
||||||
#endif // TENSORFLOW_CORE_FRAMEWORK_VARIANT_TENSOR_DATA_H_
|
#endif // TENSORFLOW_CORE_FRAMEWORK_VARIANT_TENSOR_DATA_H_
|
||||||
|
Loading…
x
Reference in New Issue
Block a user