| 96 | } |
| 97 | |
| 98 | Status ClIndirectConv2d::validate(const ITensorInfo *src, |
| 99 | const ITensorInfo *weights, |
| 100 | const ITensorInfo *biases, |
| 101 | const ITensorInfo *dst, |
| 102 | const PadStrideInfo &conv_info, |
| 103 | const ActivationLayerInfo &act_info) |
| 104 | { |
| 105 | // Initialize the direct convolution descriptor |
| 106 | const DirectConvComputeKernelInfo desc = config_indirect_convolution_nhwc(src, weights, conv_info); |
| 107 | |
| 108 | TensorShape ind_buffer_shape = misc::shape_calculator::compute_indirect_buffer_shape( |
| 109 | src->tensor_shape(), src->data_layout(), weights->tensor_shape(), conv_info, desc); |
| 110 | |
| 111 | TensorInfo indirect_buffer(ind_buffer_shape, 1, DataType::S32); |
| 112 | |
| 113 | ARM_COMPUTE_RETURN_ON_ERROR(kernels::ClIndirectConv2dAddressPrecalculationKernel::validate( |
| 114 | src, weights, &indirect_buffer, conv_info, desc)); |
| 115 | ARM_COMPUTE_RETURN_ON_ERROR(kernels::ClIndirectConv2dKernel::validate(src, weights, biases, &indirect_buffer, dst, |
| 116 | conv_info, act_info, desc)); |
| 117 | |
| 118 | return Status{}; |
| 119 | } |
| 120 | |
| 121 | void ClIndirectConv2d::run(ITensorPack &tensors) |
| 122 | { |
nothing calls this directly
no test coverage detected