diff options
Diffstat (limited to 'tests/TensorLibrary.h')
-rw-r--r-- | tests/TensorLibrary.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/TensorLibrary.h b/tests/TensorLibrary.h index b3974836ae..bdf91c6eda 100644 --- a/tests/TensorLibrary.h +++ b/tests/TensorLibrary.h @@ -73,6 +73,12 @@ public: /** Seed that is used to fill tensors with random values. */ std::random_device::result_type seed() const; + /** Provides a tensor shape for the specified image. + * + * @param[in] name Image file used to look up the raw tensor. + */ + TensorShape get_image_shape(const std::string &name); + /** Creates an uninitialised raw tensor with the given @p shape, @p * data_type and @p num_channels. * |