Home
last modified time | relevance | path

Searched refs:winograd_input_transform (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuWinogradConv2dKernel.cpp40 const ITensor *winograd_input_transform = tensors.get_const_tensor(TensorType::ACL_DST); in run_op() local
53 …win_transf_ptr = reinterpret_cast<void *>(winograd_input_transform->buffer() + winograd_input_… in run_op()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DWinograd.cpp236 SimpleTensor<T> winograd_input_transform(const SimpleTensor<T> &in, const TensorShape &output_shape… in winograd_input_transform() function
578 template SimpleTensor<float> winograd_input_transform(const SimpleTensor<float> &in, const TensorSh…
581 template SimpleTensor<half> winograd_input_transform(const SimpleTensor<half> &in, const TensorShap…
H A DWinograd.h48 SimpleTensor<T> winograd_input_transform(const SimpleTensor<T> &in, const TensorShape &output_shape…
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DWinogradConvolutionLayerFixture.h228 …SimpleTensor<T1> input_transform_out = reference::winograd_input_transform<T1>(src_t1, input_trans… in compute_reference()
346 return reference::winograd_input_transform<T>(src, output_shape, winograd_info); in compute_reference()