24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_CONVERT_FULLYCONNECTED_WEIGHTS_H
25 #define ARM_COMPUTE_CPU_CONVERT_FULLYCONNECTED_WEIGHTS_H
void configure(const ITensorInfo *src, ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout)
Configure operator for a given list of arguments.
SimpleTensor< float > src
DataLayout
[DataLayout enum definition]
constexpr auto data_layout
Basic interface for functions which have a single async CPU kernel.
Basic function to run kernels::CpuConvertFullyConnectedWeightsKernel.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const TensorShape &original_src_shape, DataLayout data_layout)
Static function to check if given info will lead to a valid configuration.
Copyright (c) 2017-2024 Arm Limited.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Store the tensor's metadata.