40 references to AvxIntrinsics
Microsoft.ML.CpuMath (15)
CpuMathUtils.netcoreapp.cs (15)
49AvxIntrinsics.MatMul(matrix, source, destination, stride, source.Size); 54AvxIntrinsics.MatMulTran(matrix, source, destination, destination.Size, stride); 134AvxIntrinsics.MatMulP(matrix, rgposSrc, sourceValues, posMin, iposMin, iposLimit, destination, stride, sourceValues.Size); 180AvxIntrinsics.ScaleAddU(scale, addend, destination); 217AvxIntrinsics.AddScaleSU(scale, source, indices, destination, count); 253AvxIntrinsics.AddSU(source, indices, destination, count); 283return (mean == 0) ? AvxIntrinsics.SumSqU(source) : AvxIntrinsics.SumSqDiffU(mean, source); 313return (mean == 0) ? AvxIntrinsics.SumAbsU(source) : AvxIntrinsics.SumAbsDiffU(mean, source); 346return AvxIntrinsics.MaxAbsU(source); 380return AvxIntrinsics.MaxAbsDiffU(mean, source); 419return AvxIntrinsics.DotSU(left, right, indices, count); 550AvxIntrinsics.SdcaL1UpdateU(primalUpdate, count, source, threshold, v, w); 593AvxIntrinsics.SdcaL1UpdateSU(primalUpdate, count, source, indices, threshold, v, w);
Microsoft.ML.CpuMath.PerformanceTests (25)
AvxPerformanceTests.cs (25)
18=> AvxIntrinsics.AddScalarU(DefaultScale, new Span<float>(dst, 0, Length)); 22=> AvxIntrinsics.Scale(DefaultScale, new Span<float>(dst, 0, Length)); 26=> AvxIntrinsics.ScaleSrcU(DefaultScale, src, dst, Length); 30=> AvxIntrinsics.ScaleAddU(DefaultScale, DefaultScale, new Span<float>(dst, 0, Length)); 35=> AvxIntrinsics.AddScaleU(DefaultScale, src, dst, Length); 40=> AvxIntrinsics.AddScaleSU(DefaultScale, src, idx, dst, IndexLength); 45=> AvxIntrinsics.AddScaleCopyU(DefaultScale, src, dst, result, Length); 49=> AvxIntrinsics.AddU(src, dst, Length); 53=> AvxIntrinsics.AddSU(src, idx, dst, IndexLength); 57=> AvxIntrinsics.MulElementWiseU(src1, src2, dst, Length); 61=> AvxIntrinsics.Sum(new Span<float>(src, 0, Length)); 66=> AvxIntrinsics.SumSqU(new Span<float>(src, 0, Length)); 71=> AvxIntrinsics.SumSqDiffU(DefaultScale, new Span<float>(src, 0, Length)); 75=> AvxIntrinsics.SumAbsU(new Span<float>(src, 0, Length)); 79=> AvxIntrinsics.SumAbsDiffU(DefaultScale, new Span<float>(src, 0, Length)); 83=> AvxIntrinsics.MaxAbsU(new Span<float>(src, 0, Length)); 87=> AvxIntrinsics.MaxAbsDiffU(DefaultScale, new Span<float>(src, 0, Length)); 92=> AvxIntrinsics.DotU(src, dst, Length); 97=> AvxIntrinsics.DotSU(src, dst, idx, IndexLength); 102=> AvxIntrinsics.Dist2(src, dst, Length); 107=> AvxIntrinsics.SdcaL1UpdateU(DefaultScale, Length, src, DefaultScale, dst, result); 112=> AvxIntrinsics.SdcaL1UpdateSU(DefaultScale, IndexLength, src, idx, DefaultScale, dst, result); 117=> AvxIntrinsics.MatMul(testMatrixAligned, testSrcVectorAligned, testDstVectorAligned, matrixLength, matrixLength); 121=> AvxIntrinsics.MatMulTran(testMatrixAligned, testSrcVectorAligned, testDstVectorAligned, matrixLength, matrixLength); 126=> AvxIntrinsics.MatMulP(testMatrixAligned, matrixIdx, testSrcVectorAligned, 0, 0, MatrixIndexLength, testDstVectorAligned, matrixLength, matrixLength);