3 writes to LatentDimAligned
Microsoft.ML.StandardTrainers (3)
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (3)
83
LatentDimAligned
= FieldAwareFactorizationMachineUtils.GetAlignedVectorLength(latentDim);
118
LatentDimAligned
= FieldAwareFactorizationMachineUtils.GetAlignedVectorLength(latentDim);
148
LatentDimAligned
= FieldAwareFactorizationMachineUtils.GetAlignedVectorLength(latentDim);
18 references to LatentDimAligned
Microsoft.ML.StandardTrainers (18)
FactorizationMachine\FieldAwareFactorizationMachineModelParameters.cs (17)
84
Host.Assert(Utils.Size(latentWeights) == checked(featureCount * fieldCount *
LatentDimAligned
));
92
_latentWeightsAligned = new AlignedArray(FeatureCount * FieldCount *
LatentDimAligned
, 16);
99
int indexAligned = j * FieldCount *
LatentDimAligned
+ f *
LatentDimAligned
;
100
for (int k = 0; k <
LatentDimAligned
; k++)
119
Host.Assert(latentWeightsAligned.Size == checked(featureCount * fieldCount *
LatentDimAligned
));
149
Host.Check(checked(featureCount * fieldCount *
LatentDimAligned
) <= Utils.ArrayMaxSize, "Latent dimension too large");
160
_latentWeightsAligned = new AlignedArray(FeatureCount * FieldCount *
LatentDimAligned
, 16);
166
int vBiasAligned = j * FieldCount *
LatentDimAligned
+ f *
LatentDimAligned
;
167
for (int k = 0; k <
LatentDimAligned
; k++)
206
Host.Assert(_latentWeightsAligned.Size == FeatureCount * FieldCount *
LatentDimAligned
);
219
int vBiasAligned = j * FieldCount *
LatentDimAligned
+ f *
LatentDimAligned
;
234
FieldAwareFactorizationMachineInterface.CalculateIntermediateVariables(FieldCount,
LatentDimAligned
, count,
276
int indexAligned = j * FieldCount *
LatentDimAligned
+ f *
LatentDimAligned
;
FactorizationMachine\FieldAwareFactorizationMachineUtils.cs (1)
101
var latentSum = new AlignedArray(_pred.FieldCount * _pred.FieldCount * _pred.
LatentDimAligned
, 16);