Searched defs:log_prob (Results 1 – 3 of 3) sorted by relevance
61 Tensor** log_prob, OpOutputList* decoded_indices, in ValidateInputsGenerateOutputs()185 Tensor* log_prob = nullptr; in Compute() local282 Tensor* log_prob = nullptr; in Compute() local
784 def log_prob(self, value, name="log_prob"): member in Distribution
4196 const float log_prob = scale * input_data[j] - precomputed; in LogSoftmax() local