Lines Matching refs:ain
198 * Invoke the kernel and apply the matrix to each cell of ain and copy to
201 * @param ain Input allocation
204 public void forEach(Allocation ain, Allocation aout) {
205 forEach(0, ain, aout, null);
220 * @param ain Input allocation
224 public void forEach(Allocation ain, Allocation aout, Script.LaunchOptions opt) {
225 if (!ain.getElement().isCompatible(Element.U8(mRS)) &&
226 !ain.getElement().isCompatible(Element.U8_2(mRS)) &&
227 !ain.getElement().isCompatible(Element.U8_3(mRS)) &&
228 !ain.getElement().isCompatible(Element.U8_4(mRS)) &&
229 !ain.getElement().isCompatible(Element.F32(mRS)) &&
230 !ain.getElement().isCompatible(Element.F32_2(mRS)) &&
231 !ain.getElement().isCompatible(Element.F32_3(mRS)) &&
232 !ain.getElement().isCompatible(Element.F32_4(mRS))) {
249 forEach(0, ain, aout, null, opt);