24.02.1
|
Go to the documentation of this file.
53 auto k = std::make_unique<kernels::ClSoftmaxKernel>();
56 _tmp_info = k->tmp_tensor_info();
58 _kernel = std::move(k);
60 _aux_mem[InternalTensorIdx::TMP] =
Descriptor used by the softmax kernels.
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
void add_tensor(int id, ITensor *tensor)
Add tensor to the pack.
static Status validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &info)
Static function to check if given info will lead to a valid configuration.
static Status validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &info)
Check if the kernel arguments are valid.
void configure(const CLCompileContext &compile_context, const ITensorInfo &src, ITensorInfo &dst, const SoftmaxKernelInfo &info)
Configure the operator.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
size_t total_size() const override
Returns the total size of the tensor in bytes.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
Tensor handler to wrap and handle tensor allocations on workspace buffers.
static CLScheduler & get()
Access the scheduler singleton.
int offset_int_vec(int offset)
Copyright (c) 2017-2024 Arm Limited.
void enqueue_op(ICLKernel &kernel, ITensorPack &tensors, bool flush=true)
Schedule the execution of the passed kernel if possible.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
#define ARM_COMPUTE_LOG_PARAMS(...)