Searched refs:_permute_input (Results 1 – 10 of 10) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEInstanceNormalizationLayer.cpp | 37 …td::move(memory_manager)), _normalization_kernel(), _is_nchw(false), _permute_input(), _permute_ou… in NEInstanceNormalizationLayer()
59 _permute_input.configure(input, &_permuted_input, PermutationVector(1U, 2U, 0U)); in configure()
89 _permute_input.run(); in run()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuDepthwiseConv2d.h | 136 std::unique_ptr<CpuPermute> _permute_input{ nullptr };
195 std::unique_ptr<CpuPermute> _permute_input{ nullptr };
|
| D | CpuDepthwiseConv2d.cpp | 99 _permute_input = std::make_unique<cpu::CpuPermute>(); in configure()
108 _permute_input->configure(src, input_perm.get(), PermutationVector(2U, 0U, 1U)); in configure()
165 _permute_input->run(pack); in run()
301 _permute_input = std::make_unique<cpu::CpuPermute>(); in configure()
304 _permute_input->configure(src, input_perm.get(), PermutationVector(2U, 0U, 1U)); in configure()
388 _permute_input->run(pack); in run()
|
| D | CpuSoftmax.cpp | 45 : _permute_input(), in CpuSoftmaxGeneric()
72 …_permute_input.configure(src, &_input_permuted, softmax_helpers::get_permutation_vector_from_softm… in configure()
177 _permute_input.run(permute_in_pack); in run()
|
| D | CpuSoftmax.h | 93 CpuPermute _permute_input; variable
|
| D | CpuWinogradConv2d.h | 115 std::unique_ptr<CpuPermute> _permute_input; variable
|
| D | CpuWinogradConv2d.cpp | 140 _permute_input(std::make_unique<CpuPermute>()), in CpuWinogradConv2d()
237 _permute_input->configure(src, &_input_nhwc, PermutationVector(2U, 0U, 1U)); in configure()
338 _permute_input->run(pack); in run()
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClSoftmax.cpp | 42 : _permute_input(std::make_unique<ClPermute>()), in ClSoftmax()
70 _permute_input->configure(compile_context, &src, &_permuted_src_info, perm_info); in configure()
150 _permute_input.get()->run(pack); in run()
|
| D | ClSoftmax.h | 78 std::unique_ptr<ClPermute> _permute_input; variable
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEInstanceNormalizationLayer.h | 101 NEPermute _permute_input; variable
|