#include <CLSplitVEx.h>

Collaboration diagram for arm_compute::CLSplitVEx:

Public Member Functions
	CLSplitVEx ()

void	configure (const ICLTensor input, const ICLTensor size_splits, uint32_t split_dim, const std::vector< ICLTensor * > &outputs, unsigned int num_splits)

void	run () override

Detailed Description

Basic function to run CLSplitVKernel

Definition at line 57 of file CLSplitVEx.h.

Constructor & Destructor Documentation

◆ CLSplitVEx()

CLSplitVEx::CLSplitVEx ( )

Default constructor

Definition at line 156 of file CLSplitVEx.cpp.

  : _input(nullptr), _size_splits(nullptr), _outputs(), _num_splits(0), _slice_functions()
{
}

Member Function Documentation

◆ configure()

void CLSplitVEx::configure	(	const ICLTensor *	input,
		const ICLTensor *	size_splits,
		uint32_t	split_dim,
		const std::vector< ICLTensor * > &	outputs,
		unsigned int	num_splits
	)

Configure the split CL kernel

Parameters

[in]	input	The input tensor to split. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32
[in]	size_splits	A 1-D tensor containing the number of tensor values per split
[out]	outputs	A vector containing the output tensor. Data types supported: Same as `input` The output tensors should match the input tensor dimensions for all shape dimensions apart from the split dimension.
[in]	split_dim	Integer value representing the input tensor dimension along which to split
[in]	num_splits	Number of splits

Definition at line 161 of file CLSplitVEx.cpp.

{
  ARM_COMPUTE_ERROR_ON_NULLPTR(input, size_splits);
  ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(size_splits, outputs, num_splits));
 
  _input = input;
  _size_splits = size_splits;
  _outputs = outputs;
  _num_splits = num_splits;
 
  // Create tensor slices
  _slice_functions.resize(_num_splits);
 
  // Extract output tensor info
  std::vector<ITensorInfo *> outputs_info;
  for (auto &&output : _outputs)
  {
    ARM_COMPUTE_ERROR_ON_NULLPTR(output);
    outputs_info.emplace_back(output->info());
  }
 
  // Validate slices
  ARM_COMPUTE_ERROR_THROW_ON(validate_slices(_input->info(), outputs_info, split_dim));
 
  // Configure slices
  configure_slices(_input, _outputs, _slice_functions, split_dim);
}

◆ run()

void CLSplitVEx::run ( )

override

Definition at line 190 of file CLSplitVEx.cpp.

{
  // execute the slices
  for (unsigned i = 0; i < _outputs.size(); ++i)
  {
    _slice_functions[i].run();
  }
}

Referenced by package.infer.session::inference().

The documentation for this class was generated from the following files: