Searched refs:ComputeLogitsHelper (Results 1 – 2 of 2) sorted by relevance
40 TensorView<float> ComputeLogitsHelper(const int input_index_features,75 return ComputeLogitsHelper(kInputIndexFeatures, kOutputIndexLogits, in ComputeLogits()
125 TensorView<float> ComputeLogitsHelper(const int input_index_features, in ComputeLogitsHelper() function