Searched refs:op_tape (Results 1 – 1 of 1) sorted by relevance
506 OpTape<BackwardFunction, TapeTensor> op_tape; member525 OpTape<BackwardFunction, TapeTensor>* op_tape, in PrepareBackprop() argument541 auto op_it = op_tape->find(op_id); in PrepareBackprop()542 auto result_op_it = result.op_tape.find(op_id); in PrepareBackprop()543 if (op_id == -1 || op_it == op_tape->end() || in PrepareBackprop()544 result_op_it != result.op_tape.end()) { in PrepareBackprop()547 CHECK(result.op_tape.emplace(op_id, op_it->second).second); in PrepareBackprop()560 op_tape->erase(op_it); in PrepareBackprop()574 for (const auto& op_pair : *op_tape) { in PrepareBackprop()578 op_tape->clear(); in PrepareBackprop()[all …]