From 5ff38da7e18e91243a7f6b8e642f8b40f5846068 Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Tue, 2 Mar 2021 09:41:13 +0000 Subject: Create ClPRelu operator Make the class that was in experimental namespace as ClOperator to prepare porting to new interface. The followings are added as a part of this change Also, in-place computation is now correctly considered to be aligned with the class description. Test cases to test in-place computation are added. Partially Implements: COMPMID-4184 Signed-off-by: Sang-Hoon Park Change-Id: I71c18ab47fe0370a2060d5303a58ff3650c0093f Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5201 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio --- tests/Utils.h | 23 +++++++++++++++++------ 1 file changed, 17 insertions(+), 6 deletions(-) (limited to 'tests/Utils.h') diff --git a/tests/Utils.h b/tests/Utils.h index ab256032b0..2569c41a9e 100644 --- a/tests/Utils.h +++ b/tests/Utils.h @@ -512,6 +512,21 @@ inline bool is_in_valid_region(const ValidRegion &valid_region, Coordinates coor return true; } +/** Create and initialize a tensor of the given type. + * + * @param[in] info Tensor information to be used to create the tensor + * @param[in] ctx (Optional) Pointer to the runtime context. + * + * @return Initialized tensor of given type. + */ +template +inline T create_tensor(const TensorInfo &info, IRuntimeContext *ctx = nullptr) +{ + T tensor(ctx); + tensor.allocator()->init(info); + return tensor; +} + /** Create and initialize a tensor of the given type. * * @param[in] shape Tensor shape. @@ -531,9 +546,8 @@ inline T create_tensor(const TensorShape &shape, DataType data_type, int num_cha TensorInfo info(shape, num_channels, data_type); info.set_quantization_info(quantization_info); info.set_data_layout(data_layout); - tensor.allocator()->init(info); - return tensor; + return create_tensor(info, ctx); } /** Create and initialize a tensor of the given type. @@ -549,10 +563,7 @@ inline T create_tensor(const TensorShape &shape, Format format, IRuntimeContext { TensorInfo info(shape, format); - T tensor(ctx); - tensor.allocator()->init(info); - - return tensor; + return create_tensor(info, ctx); } /** Create and initialize a multi-image of the given type. -- cgit v1.2.1