Searched defs:head_output (Results 1 – 2 of 2) sorted by relevance
158 const Tensor head_output = head(input); in forward() local172 const Tensor& head_output) { in _get_full_log_prob()196 const Tensor head_output = head(input); in log_prob() local201 const Tensor head_output = head(input); in predict() local
261 def _get_full_log_prob(self, input, head_output): argument