Searched refs:history_map (Results 1 – 2 of 2) sorted by relevance
111 real_pred = grad_state.history_map.get(pred.name)122 grad_state.history_map[pred.name] = real_pred
936 def history_map(self): member in GradLoopState1099 pred = grad_state.history_map.get(cond_ctxt.pred.name)1367 pred = grad_state.history_map.get(op_ctxt.pred.name)