Searched refs:permutation_vector (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NESoftmaxLayer.cpp | 141 …const PermutationVector permutation_vector = softmax_helpers::get_permutation_vector_from_softmax_… in validate() local
142 …d_shape = misc::shape_calculator::compute_permutation_output_shape(*input, permutation_vector); in validate()
144 … ARM_COMPUTE_RETURN_ON_ERROR(NEPermute::validate(input, &input_permuted, permutation_vector)); in validate()
146 … ARM_COMPUTE_RETURN_ON_ERROR(NEPermute::validate(&output_permuted, output, permutation_vector)); in validate()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLSoftmaxLayer.cpp | 135 …const PermutationVector permutation_vector = softmax_helpers::get_permutation_vector_from_softmax_… in validate() local
136 …d_shape = misc::shape_calculator::compute_permutation_output_shape(*input, permutation_vector); in validate()
138 … ARM_COMPUTE_RETURN_ON_ERROR(CLPermute::validate(input, &input_permuted, permutation_vector)); in validate()
140 … ARM_COMPUTE_RETURN_ON_ERROR(CLPermute::validate(&output_permuted, output, permutation_vector)); in validate()
|
| /external/ComputeLibrary/arm_compute/graph/nodes/ |
| D | PermuteLayerNode.h | 48 const PermutationVector &permutation_vector() const;
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | PermuteLayerNode.cpp | 42 const PermutationVector &PermuteLayerNode::permutation_vector() const in permutation_vector() function in arm_compute::graph::PermuteLayerNode
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_types.i | 9 %include <typemaps/permutation_vector.i>
|
| /external/ComputeLibrary/arm_compute/graph/backends/ |
| D | ValidateHelpers.h | 437 const PermutationVector &perm = node.permutation_vector(); in validate_permute_layer()
|
| D | FunctionHelpers.h | 1274 const PermutationVector &perm = node.permutation_vector(); in create_permute_layer()
|