Lines Matching refs:ains
515 bool RsdCpuScriptImpl::forEachMtlsSetup(const Allocation ** ains,
525 const Allocation* ain = ains[index];
546 const Allocation *ain0 = ains[0];
554 if (!ain0->hasSameDims(ains[Index])) {
581 if (!ains[0]->hasSameDims(aout)) {
679 if (ains) {
680 memcpy(mtls->ains, ains, inLen * sizeof(ains[0]));
693 mtls->fep.inPtr[index] = (const uint8_t*)ains[index]->mHal.drvState.lod[0].mallocPtr;
694 mtls->fep.inStride[index] = ains[index]->getType()->getElementSizeBytes();
709 const Allocation ** ains,
718 if (forEachMtlsSetup(ains, inLen, aout, usr, usrLen, sc, &mtls)) {
722 mCtx->launchThreads(ains, inLen, aout, sc, &mtls);
933 void RsdCpuScriptImpl::preLaunch(uint32_t slot, const Allocation ** ains,
938 void RsdCpuScriptImpl::postLaunch(uint32_t slot, const Allocation ** ains,
