Basic function to run @ref opencl::kernels::ClComplexMulKernel */
| 91 | |
| 92 | /** Basic function to run @ref opencl::kernels::ClComplexMulKernel */ |
| 93 | class ClComplexMul : public IClOperator |
| 94 | { |
| 95 | public: |
| 96 | /** Initialise the kernel's sources, dst. |
| 97 | * |
| 98 | * @param[in] compile_context The compile context to be used. |
| 99 | * @param[in, out] src1 An src tensor info. Data types supported: F16/F32. Number of channels supported: 2. |
| 100 | * The src tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. |
| 101 | * @param[in, out] src2 An src tensor info. Data types supported: same as @p src1. Number of channels supported: same as @p src1. |
| 102 | * The src tensor is [in, out] because its TensorInfo might be modified inside the kernel in case of broadcasting of dimension 0. |
| 103 | * @param[out] dst The dst tensor info, Data types supported: same as @p src1. Number of channels supported: same as @p src1. |
| 104 | * @param[in] act_info (Optional) Activation layer information in case of a fused activation. |
| 105 | */ |
| 106 | void configure(const CLCompileContext &compile_context, |
| 107 | ITensorInfo *src1, |
| 108 | ITensorInfo *src2, |
| 109 | ITensorInfo *dst, |
| 110 | const ActivationLayerInfo &act_info = ActivationLayerInfo()); |
| 111 | /** Static function to check if given info will lead to a valid configuration |
| 112 | * |
| 113 | * Similar to @ref ClComplexMul::configure() |
| 114 | * |
| 115 | * @return a status |
| 116 | */ |
| 117 | static Status validate(const ITensorInfo *src1, |
| 118 | const ITensorInfo *src2, |
| 119 | const ITensorInfo *dst, |
| 120 | const ActivationLayerInfo &act_info = ActivationLayerInfo()); |
| 121 | }; |
| 122 | } // namespace opencl |
| 123 | } // namespace arm_compute |
| 124 | #endif /* ARM_COMPUTE_CL_MUL_H */ |
nothing calls this directly
no test coverage detected