Searched refs:_softmax_kernel (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NESoftmaxLayer.cpp | 42 …emory_manager)), _permute_input(), _permute_output(), _max_kernel(), _softmax_kernel(), _fill_bord… in NESoftmaxLayerGeneric()
87 _softmax_kernel = arm_compute::support::cpp14::make_unique<NELogits1DSoftmaxKernel<IS_LOG>>(); in configure()
95 _softmax_kernel->configure(tmp_input, &_max, &_output_permuted, beta, &_tmp); in configure()
109 _softmax_kernel->configure(tmp_input, &_max, output, beta, &_tmp); in configure()
170 NEScheduler::get().schedule(_softmax_kernel.get(), Window::DimY); in run()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NESoftmaxLayer.h | 103 std::unique_ptr<NELogits1DSoftmaxKernel<IS_LOG>> _softmax_kernel; variable
|