24 #ifndef ARM_COMPUTE_TEST_SPACE_TO_BATCH_LAYER_H 25 #define ARM_COMPUTE_TEST_SPACE_TO_BATCH_LAYER_H 39 SimpleTensor<T>
space_to_batch(
const SimpleTensor<T> &
src,
const SimpleTensor<int32_t> &block_shape,
const SimpleTensor<int32_t> &paddings,
const TensorShape &
dst_shape);
SimpleTensor< T > space_to_batch(const SimpleTensor< T > &src, const SimpleTensor< int32_t > &block_shape, const SimpleTensor< int32_t > &paddings, const TensorShape &dst_shape)
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.