Lines Matching refs:aout
797 void RsdCpuScriptImpl::forEachMtlsSetup(const Allocation * ain, Allocation * aout, in forEachMtlsSetup() argument
809 if (aout && (const uint8_t *)aout->mHal.drvState.lod[0].mallocPtr == NULL) { in forEachMtlsSetup()
821 } else if (aout != NULL) { in forEachMtlsSetup()
822 const Type *outType = aout->getType(); in forEachMtlsSetup()
833 if (ain != NULL && aout != NULL) { in forEachMtlsSetup()
834 if (!ain->hasSameDims(aout)) { in forEachMtlsSetup()
884 mtls->aout = aout; in forEachMtlsSetup()
902 if (aout) { in forEachMtlsSetup()
903 mtls->fep.ptrOut = (uint8_t *)aout->mHal.drvState.lod[0].mallocPtr; in forEachMtlsSetup()
904 mtls->fep.eStrideOut = aout->getType()->getElementSizeBytes(); in forEachMtlsSetup()
905 mtls->fep.yStrideOut = aout->mHal.drvState.lod[0].stride; in forEachMtlsSetup()
910 Allocation * aout, in forEachMtlsSetup() argument
929 if (aout && (const uint8_t *)aout->mHal.drvState.lod[0].mallocPtr == NULL) { in forEachMtlsSetup()
951 } else if (aout != NULL) { in forEachMtlsSetup()
952 const Type *outType = aout->getType(); in forEachMtlsSetup()
963 if (ains != NULL && aout != NULL) { in forEachMtlsSetup()
964 if (!ains[0]->hasSameDims(aout)) { in forEachMtlsSetup()
1014 mtls->aout = aout; in forEachMtlsSetup()
1043 if (aout) { in forEachMtlsSetup()
1044 mtls->fep.ptrOut = (uint8_t *)aout->mHal.drvState.lod[0].mallocPtr; in forEachMtlsSetup()
1045 mtls->fep.eStrideOut = aout->getType()->getElementSizeBytes(); in forEachMtlsSetup()
1046 mtls->fep.yStrideOut = aout->mHal.drvState.lod[0].stride; in forEachMtlsSetup()
1053 Allocation * aout, in invokeForEach() argument
1059 forEachMtlsSetup(ain, aout, usr, usrLen, sc, &mtls); in invokeForEach()
1063 mCtx->launchThreads(ain, aout, sc, &mtls); in invokeForEach()
1070 Allocation * aout, in invokeForEachMulti() argument
1077 forEachMtlsSetup(ains, inLen, aout, usr, usrLen, sc, &mtls); in invokeForEachMulti()
1081 mCtx->launchThreads(ains, inLen, aout, sc, &mtls); in invokeForEachMulti()
1347 Allocation * aout, const void * usr, in preLaunch() argument
1353 Allocation * aout, const void * usr, in postLaunch() argument