Home
last modified time | relevance | path

Searched refs:PermutationVector (Results 1 – 25 of 173) sorted by relevance

1234567

/external/ComputeLibrary/src/cpu/kernels/
DCpuPermuteKernel.cpp49 inline bool is_permutation_supported(const PermutationVector &v) in is_permutation_supported()
51 static const std::array<PermutationVector, 2> permutations2 = in is_permutation_supported()
54 PermutationVector(0U, 1U), in is_permutation_supported()
55 PermutationVector(1U, 0U), in is_permutation_supported()
58 static const std::array<PermutationVector, 6> permutations3 = in is_permutation_supported()
61 PermutationVector(2U, 0U, 1U), in is_permutation_supported()
62 PermutationVector(1U, 2U, 0U), in is_permutation_supported()
63 PermutationVector(0U, 1U, 2U), in is_permutation_supported()
64 PermutationVector(0U, 2U, 1U), in is_permutation_supported()
65 PermutationVector(1U, 0U, 2U), in is_permutation_supported()
[all …]
DCpuPermuteKernel.h50 void configure(const ITensorInfo *src, ITensorInfo *dst, const PermutationVector &perm);
57 …static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const PermutationVector &pe…
64 PermutationVector _perm{};
/external/ComputeLibrary/tests/validation/NEON/
DPermute.cpp47 PermutationVector(0U, 1U),
48 PermutationVector(1U, 0U),
52 PermutationVector(2U, 0U, 1U),
53 PermutationVector(1U, 2U, 0U),
54 PermutationVector(0U, 1U, 2U),
55 PermutationVector(0U, 2U, 1U),
56 PermutationVector(1U, 0U, 2U),
57 PermutationVector(2U, 1U, 0U),
61 PermutationVector(3U, 2U, 0U, 1U),
62 PermutationVector(3U, 2U, 1U, 0U),
[all …]
/external/ComputeLibrary/tests/validation/CL/
DPermute.cpp47 PermutationVector(2U, 0U, 1U),
48 PermutationVector(1U, 2U, 0U),
49 PermutationVector(0U, 1U, 2U),
50 PermutationVector(0U, 2U, 1U),
51 PermutationVector(1U, 0U, 2U),
52 PermutationVector(2U, 1U, 0U),
56 PermutationVector(3U, 2U, 0U, 1U),
57 PermutationVector(3U, 2U, 1U, 0U),
58 PermutationVector(2U, 3U, 1U, 0U),
59 PermutationVector(1U, 3U, 2U, 0U),
[all …]
/external/ComputeLibrary/tests/validation/reference/
DPermute.cpp38 SimpleTensor<T> permute(const SimpleTensor<T> &src, PermutationVector perm) in permute()
61 template SimpleTensor<int8_t> permute(const SimpleTensor<int8_t> &src, PermutationVector perm);
62 template SimpleTensor<uint8_t> permute(const SimpleTensor<uint8_t> &src, PermutationVector perm);
63 template SimpleTensor<int16_t> permute(const SimpleTensor<int16_t> &src, PermutationVector perm);
64 template SimpleTensor<uint16_t> permute(const SimpleTensor<uint16_t> &src, PermutationVector perm);
65 template SimpleTensor<uint32_t> permute(const SimpleTensor<uint32_t> &src, PermutationVector perm);
66 template SimpleTensor<float> permute(const SimpleTensor<float> &src, PermutationVector perm);
67 template SimpleTensor<half> permute(const SimpleTensor<half> &src, PermutationVector perm);
DConcatenateLayer.cpp128 t = reference::permute<T>(t, PermutationVector(1U, 0U)); in concatenate_layer()
130 dst = reference::permute<T>(dst, PermutationVector(1U, 0U)); in concatenate_layer()
131 … return reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(1U, 0U)); in concatenate_layer()
137 t = reference::permute<T>(t, PermutationVector(2U, 1U, 0U)); in concatenate_layer()
139 dst = reference::permute<T>(dst, PermutationVector(2U, 1U, 0U)); in concatenate_layer()
140 … return reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(2U, 1U, 0U)); in concatenate_layer()
146 t = reference::permute<T>(t, PermutationVector(3U, 2U, 1U, 0U)); in concatenate_layer()
148 dst = reference::permute<T>(dst, PermutationVector(3U, 2U, 1U, 0U)); in concatenate_layer()
149 …auto ret = reference::permute<T>(widthconcatenate_layer(srcs, dst), PermutationVector(3U, 2U, 1U, … in concatenate_layer()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_types.i33 ") PermutationVector;
35 class PermutationVector
42 PermutationVector(const ValueType *dimMappings, SizeType numDimMappings);
64 bool IsInverse(const PermutationVector& other);
67 %extend PermutationVector {
73 bool __eq__(PermutationVector other) { in __eq__()
107 %ignore PermutationVector;
/external/ComputeLibrary/src/core/helpers/
DSoftmaxHelpers.cpp30 PermutationVector get_permutation_vector_from_softmax_axis(size_t axis) in get_permutation_vector_from_softmax_axis()
35 return PermutationVector(1U, 0U, 2U, 3U); in get_permutation_vector_from_softmax_axis()
37 return PermutationVector(2U, 1U, 0U, 3U); in get_permutation_vector_from_softmax_axis()
39 return PermutationVector(3U, 1U, 2U, 0U); in get_permutation_vector_from_softmax_axis()
/external/ComputeLibrary/tests/validation/CPP/
DPermute.cpp47 PermutationVector(2U, 0U, 1U),
48 PermutationVector(1U, 2U, 0U),
49 PermutationVector(0U, 1U, 2U),
50 PermutationVector(0U, 2U, 1U),
51 PermutationVector(1U, 0U, 2U),
52 PermutationVector(2U, 1U, 0U),
/external/armnn/src/armnn/test/
DUtilsTests.cpp211 armnn::PermutationVector mappings({ 0, 3, 2, 1 });
226 armnn::PermutationVector pv = armnn::PermutationVector({});
232 armnn::PermutationVector pv = armnn::PermutationVector({ 1u, 2u, 0u });
240 armnn::PermutationVector pv = armnn::PermutationVector({ 2u, 1u, 0u });
253 armnn::PermutationVector zeroPVector({});
258 armnn::PermutationVector fourPVector({ 0, 3, 2, 1 });
276 armnn::PermutationVector twoPVector({ 0, 1 });
/external/ComputeLibrary/src/runtime/CL/functions/
DCLDepthwiseConvolutionLayer.cpp105 …e_input_to_nhwc.configure(compile_context, input, &_permuted_input, PermutationVector(2U, 0U, 1U)); in configure()
109 …hts_to_nhwc.configure(compile_context, weights, &_permuted_weights, PermutationVector(2U, 0U, 1U)); in configure()
150 …utput_to_nchw.configure(compile_context, &_permuted_output, output, PermutationVector(1U, 2U, 0U)); in configure()
209 permute(permuted_input_shape, PermutationVector(2U, 0U, 1U)); in validate()
210 permute(permuted_weights_shape, PermutationVector(2U, 0U, 1U)); in validate()
211 permute(permuted_output_shape, PermutationVector(2U, 0U, 1U)); in validate()
217 …ARM_COMPUTE_RETURN_ON_ERROR(CLPermute::validate(input, &permuted_input, PermutationVector(2U, 0U, … in validate()
218 …UTE_RETURN_ON_ERROR(CLPermute::validate(weights, &permuted_weights, PermutationVector(2U, 0U, 1U))… in validate()
226 …ARM_COMPUTE_RETURN_ON_ERROR(CLPermute::validate(&permuted_output, output, PermutationVector(1U, 2U… in validate()
DCLPermute.cpp51 void CLPermute::configure(const ICLTensor *input, ICLTensor *output, const PermutationVector &perm) in configure()
56 …Context &compile_context, const ICLTensor *input, ICLTensor *output, const PermutationVector &perm) in configure()
68 …rmute::validate(const ITensorInfo *input, const ITensorInfo *output, const PermutationVector &perm) in validate()
/external/armnn/include/armnnUtils/
DTranspose.hpp15 …TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings);
17 …Info TransposeTensorShape(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings);
19 void Transpose(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
DPermute.hpp17 const armnn::PermutationVector& mappings);
20 const armnn::PermutationVector& mappings);
22 void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings,
/external/armnn/include/armnnTestUtils/
DDataLayoutUtils.hpp16 const armnn::PermutationVector nchwToNhwc = { 0, 3, 1, 2 }; in PermuteTensorNchwToNhwc()
28 const armnn::PermutationVector nhwcToNchw = { 0, 2, 3, 1 }; in PermuteTensorNhwcToNchw()
41 const armnn::PermutationVector ndhwcToNcdhw = { 0, 2, 3, 4, 1 }; in PermuteTensorNdhwcToNcdhw()
53 const armnn::PermutationVector ncdhwToNdhwc = { 0, 4, 1, 2, 3 }; in PermuteTensorNcdhwToNdhwc()
/external/armnn/src/backends/backendsCommon/
DWorkloadUtils.cpp19 const PermutationVector& permutationVector, void* permuteBuffer) in PermuteTensor()
127 PermutationVector permutationVector{ 3, 2, 0, 1 }; in ConvertWeightTensorInfoFromArmnnToAcl()
146 PermutationVector permutationVector{}; in Convert1HWOTensorToAcl()
188 PermutationVector permutationVector{ 0, 2, 3, 1 }; in Convert1HWOTensorInfoToAcl()
224 PermutationVector permutationVector = { 2, 3, 1, 0 }; in Convert1HWOtoMIHW()
247 PermutationVector permutationVector{}; in ConvertWeightTensorFromArmnnToAcl()
344 armnn::PermutationVector GeneratePermutationVectorOnLastTwoDimensions(unsigned int rank) in GeneratePermutationVectorOnLastTwoDimensions()
346 armnn::PermutationVector permutationVector{}; in GeneratePermutationVectorOnLastTwoDimensions()
/external/ComputeLibrary/arm_compute/graph/nodes/
DPermuteLayerNode.h43 PermuteLayerNode(PermutationVector perm, DataLayout layout = DataLayout::UNKNOWN);
48 const PermutationVector &permutation_vector() const;
57 PermutationVector _perm;
/external/ComputeLibrary/src/gpu/cl/kernels/
DClPermuteKernel.h55 …Context &compile_context, const ITensorInfo *src, ITensorInfo *dst, const PermutationVector &perm);
62 …static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const PermutationVector &pe…
68 PermutationVector _perm{};
/external/armnn/python/pyarmnn/test/
Dtest_types.py21 pv = ann.PermutationVector((0, 2, 3, 1))
25 pv2 = ann.PermutationVector((0, 2, 3, 1))
28 pv4 = ann.PermutationVector((0, 3, 1, 2))
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLPermute.h71 void configure(const ICLTensor *input, ICLTensor *output, const PermutationVector &perm);
81 …ontext &compile_context, const ICLTensor *input, ICLTensor *output, const PermutationVector &perm);
92 …tatus validate(const ITensorInfo *input, const ITensorInfo *output, const PermutationVector &perm);
/external/ComputeLibrary/arm_compute/core/CPP/kernels/
DCPPPermuteKernel.h63 void configure(const ITensor *input, ITensor *output, const PermutationVector &perm);
72 …tatus validate(const ITensorInfo *input, const ITensorInfo *output, const PermutationVector &perm);
94 PermutationVector _perm;
/external/armnn/src/armnnUtils/
DPermute.cpp23 PermuteLoop(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings) in PermuteLoop()
99 const armnn::PermutationVector& mappings) in Permuted()
116 const armnn::PermutationVector& mappings) in Permuted()
131 void Permute(const armnn::TensorShape& dstShape, const armnn::PermutationVector& mappings, in Permute()
DTranspose.cpp23 TransposeLoop(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings) in TransposeLoop()
98 … TransposeTensorShape(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings) in TransposeTensorShape()
113 …rInfo TransposeTensorShape(const armnn::TensorInfo& info, const armnn::PermutationVector& mappings) in TransposeTensorShape()
120 void Transpose(const armnn::TensorShape& srcShape, const armnn::PermutationVector& mappings, in Transpose()
/external/ComputeLibrary/src/cpu/operators/
DCpuDepthwiseConv2d.cpp108 _permute_input->configure(src, input_perm.get(), PermutationVector(2U, 0U, 1U)); in configure()
112 _permute_weights->configure(weights, weights_perm.get(), PermutationVector(2U, 0U, 1U)); in configure()
123 _permute_output->configure(output_perm.get(), dst, PermutationVector(1U, 2U, 0U)); in configure()
304 _permute_input->configure(src, input_perm.get(), PermutationVector(2U, 0U, 1U)); in configure()
308 _permute_weights->configure(weights, weights_perm.get(), PermutationVector(2U, 0U, 1U)); in configure()
321 _permute_output->configure(output_perm.get(), dst, PermutationVector(1U, 2U, 0U)); in configure()
343 permute(permuted_input_shape, PermutationVector(2U, 0U, 1U)); in validate()
344 permute(permuted_weights_shape, PermutationVector(2U, 0U, 1U)); in validate()
345 permute(permuted_output_shape, PermutationVector(2U, 0U, 1U)); in validate()
351 …ARM_COMPUTE_RETURN_ON_ERROR(CpuPermute::validate(src, &permuted_input, PermutationVector(2U, 0U, 1… in validate()
[all …]
/external/armnn/shim/sl/canonical/
DConversionUtils.hpp102 const armnn::PermutationVector& mappings);
456 const armnn::PermutationVector IdentityPermutation4D({ 0U, 1U, 2U, 3U });
457 const armnn::PermutationVector IdentityPermutation3D({ 0U, 1U, 2U });
458 const armnn::PermutationVector SwapDim2And3({ 0U, 1U, 3U, 2U });
461 const armnn::PermutationVector RotateTensorLeft({ 1U, 2U, 0U });
462 const armnn::PermutationVector RotateTensorRight({ 2U, 0U, 1U });
466 const armnn::PermutationVector& mappings) in AddTransposeLayer()
534 const armnn::PermutationVector& mapping, in SwizzleInputs()
557 const armnn::PermutationVector& mapping) in TransposeInputTensors()
596 … std::pair<armnn::PermutationVector, armnn::PermutationVector> & permutationPair) in CreateConcatPermutationParameters() argument
[all …]

1234567