Searched refs:history_map (Results 1 – 2 of 2) sorted by relevance
116 real_pred = grad_state.history_map.get(pred.name)127 grad_state.history_map[pred.name] = real_pred
264 def history_map(self): member in _GradLoopState420 pred = grad_state.history_map.get(cond_ctxt.pred.name)685 pred = grad_state.history_map.get(op_ctxt.pred.name)