| 112 | } |
| 113 | |
| 114 | Status ClDirectConv2d::validate(const ITensorInfo *src, |
| 115 | const ITensorInfo *weights, |
| 116 | const ITensorInfo *biases, |
| 117 | const ITensorInfo *dst, |
| 118 | const PadStrideInfo &conv_info, |
| 119 | const ActivationLayerInfo &act_info) |
| 120 | { |
| 121 | // Initialize the direct convolution descriptor |
| 122 | const DirectConvComputeKernelInfo desc = config_direct_convolution_nhwc(src, weights, conv_info); |
| 123 | |
| 124 | ARM_COMPUTE_RETURN_ON_ERROR( |
| 125 | kernels::ClDirectConv2dKernel::validate(src, weights, biases, dst, conv_info, ActivationLayerInfo(), desc)); |
| 126 | if (act_info.enabled()) |
| 127 | { |
| 128 | ARM_COMPUTE_RETURN_ON_ERROR(kernels::ClActivationKernel::validate(dst, dst, act_info)); |
| 129 | } |
| 130 | return Status{}; |
| 131 | } |
| 132 | |
| 133 | void ClDirectConv2d::run(ITensorPack &tensors) |
| 134 | { |
nothing calls this directly
no test coverage detected