Lines Matching refs:ains
558 bool RsdCpuScriptImpl::reduceMtlsSetup(const Allocation ** ains,
563 rsAssert(ains && (inLen >= 1) && aout);
568 if (allocationLODIsNull(ains[index])) {
581 const Allocation *ain0 = ains[0];
589 if (!ain0->hasSameDims(ains[Index])) {
617 memcpy(mtls->ains, ains, inLen * sizeof(ains[0]));
620 mtls->redp.inPtr[index] = (const uint8_t*)ains[index]->mHal.drvState.lod[0].mallocPtr;
621 mtls->redp.inStride[index] = ains[index]->getType()->getElementSizeBytes();
629 bool RsdCpuScriptImpl::forEachMtlsSetup(const Allocation ** ains,
635 if (ains == nullptr && inLen != 0) {
645 if (allocationLODIsNull(ains[index])) {
659 const Allocation *ain0 = ains[0];
667 if (!ain0->hasSameDims(ains[Index])) {
692 if (!ains[0]->hasSameDims(aout)) {
708 if (ains) {
709 memcpy(mtls->ains, ains, inLen * sizeof(ains[0]));
722 mtls->fep.inPtr[index] = (const uint8_t*)ains[index]->mHal.drvState.lod[0].mallocPtr;
723 mtls->fep.inStride[index] = ains[index]->getType()->getElementSizeBytes();
738 const Allocation ** ains,
747 if (forEachMtlsSetup(ains, inLen, aout, usr, usrLen, sc, &mtls)) {
751 mCtx->launchForEach(ains, inLen, aout, sc, &mtls);
757 const Allocation ** ains, uint32_t inLen,
762 if (reduceMtlsSetup(ains, inLen, aout, sc, &mtls)) {
765 mCtx->launchReduce(ains, inLen, aout, &mtls);
988 void RsdCpuScriptImpl::preLaunch(uint32_t slot, const Allocation ** ains,
993 void RsdCpuScriptImpl::postLaunch(uint32_t slot, const Allocation ** ains,