| 44 | } |
| 45 | |
| 46 | void ClSoftmax::configure(const CLCompileContext &compile_context, |
| 47 | const ITensorInfo &src, |
| 48 | ITensorInfo &dst, |
| 49 | const SoftmaxKernelInfo &info) |
| 50 | { |
| 51 | ARM_COMPUTE_LOG_PARAMS(src, dst, info); |
| 52 | |
| 53 | auto k = std::make_unique<kernels::ClSoftmaxKernel>(); |
| 54 | k->configure(compile_context, src, dst, info); |
| 55 | |
| 56 | _tmp_info = k->tmp_tensor_info(); |
| 57 | |
| 58 | _kernel = std::move(k); |
| 59 | |
| 60 | _aux_mem[InternalTensorIdx::TMP] = |
| 61 | MemoryInfo(offset_int_vec(InternalTensorIdx::TMP), MemoryLifetime::Temporary, _tmp_info.total_size()); |
| 62 | } |
| 63 | |
| 64 | Status ClSoftmax::validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &info) |
| 65 | { |
nothing calls this directly
no test coverage detected