| 171 | } // namespace |
| 172 | |
| 173 | void CpuCastKernel::configure(const ITensorInfo *src, ITensorInfo *dst, ConvertPolicy policy) |
| 174 | { |
| 175 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuCastKernel::configure"); |
| 176 | ARM_COMPUTE_ERROR_ON_NULLPTR(src, dst); |
| 177 | |
| 178 | // Auto initialize dst shape if not initialized (We can only auto-configure the shape, datatype must be given) |
| 179 | set_shape_if_empty(*dst, src->tensor_shape()); |
| 180 | |
| 181 | _policy = policy; |
| 182 | |
| 183 | ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(src, dst, policy)); |
| 184 | |
| 185 | // Configure kernel window |
| 186 | Window win = calculate_max_window(*src, Steps()); |
| 187 | |
| 188 | ICPPKernel::configure(win); |
| 189 | } |
| 190 | |
| 191 | Status CpuCastKernel::validate(const ITensorInfo *src, const ITensorInfo *dst, ConvertPolicy policy) |
| 192 | { |
nothing calls this directly
no test coverage detected