| 94 | } |
| 95 | |
| 96 | void NEQLSTMLayer::TensorCopyKernel::configure(ITensor &src, ITensor &dst) |
| 97 | { |
| 98 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, |
| 99 | "NEQLSTMLayer::TensorCopyKernel::configure"); |
| 100 | ARM_COMPUTE_ERROR_THROW_ON(NEQLSTMLayer::TensorCopyKernel::validate(*src.info(), *dst.info())); |
| 101 | ARM_COMPUTE_LOG_PARAMS(src, dst); |
| 102 | |
| 103 | _src = &src; |
| 104 | _dst = &dst; |
| 105 | _row_size = std::min(_src->info()->tensor_shape().x(), _dst->info()->tensor_shape().x()); |
| 106 | _window = calculate_max_window(*_src->info(), Steps()); |
| 107 | } |
| 108 | |
| 109 | void NEQLSTMLayer::TensorCopyKernel::run() |
| 110 | { |
no test coverage detected