Searched defs:loss_tensor (Results 1 – 3 of 3) sorted by relevance
46 def __init__(self, loss_tensor, tolerance): argument
735 def __init__(self, loss_tensor, fail_on_nan_loss=True): argument
1230 def __init__(self, loss_tensor, every_n_steps=100, fail_on_nan_loss=True): argument