Lines Matching refs:ain
250 const Allocation * ain, in rsdScriptInvokeForEachMtlsSetup() argument
259 if (ain) { in rsdScriptInvokeForEachMtlsSetup()
260 mtls->fep.dimX = ain->getType()->getDimX(); in rsdScriptInvokeForEachMtlsSetup()
261 mtls->fep.dimY = ain->getType()->getDimY(); in rsdScriptInvokeForEachMtlsSetup()
262 mtls->fep.dimZ = ain->getType()->getDimZ(); in rsdScriptInvokeForEachMtlsSetup()
301 rsAssert(!ain || (ain->getType()->getDimZ() == 0)); in rsdScriptInvokeForEachMtlsSetup()
305 mtls->ain = ain; in rsdScriptInvokeForEachMtlsSetup()
315 if (ain) { in rsdScriptInvokeForEachMtlsSetup()
316 DrvAllocation *aindrv = (DrvAllocation *)ain->mHal.drv; in rsdScriptInvokeForEachMtlsSetup()
318 mtls->fep.eStrideIn = ain->getType()->getElementSizeBytes(); in rsdScriptInvokeForEachMtlsSetup()
335 const Allocation * ain, in rsdScriptLaunchThreads() argument
418 const Allocation * ain, in rsdScriptInvokeForEach() argument
427 rsdScriptInvokeForEachMtlsSetup(rsc, ain, aout, usr, usrLen, sc, &mtls); in rsdScriptInvokeForEach()
444 rsdScriptLaunchThreads(rsc, s, slot, ain, aout, usr, usrLen, sc, &mtls); in rsdScriptInvokeForEach()