Searched defs:inner_losses (Results 1 – 3 of 3) sorted by relevance
90 inner_losses = [ variable
97 inner_losses = vmap(get_loss_for_task)(task[0], task[1], task[2], task[3]) variable
93 inner_losses = vmap(get_loss_for_task)(task[0], task[1], task[2], task[3]) variable