Searched refs:op_tape (Results 1 – 1 of 1) sorted by relevance
306 OpTape<BackwardFunction, TapeTensor> op_tape; member325 OpTape<BackwardFunction, TapeTensor>* op_tape, in PrepareBackprop() argument341 auto op_it = op_tape->find(op_id); in PrepareBackprop()342 auto result_op_it = result.op_tape.find(op_id); in PrepareBackprop()343 if (op_id == -1 || op_it == op_tape->end() || in PrepareBackprop()344 result_op_it != result.op_tape.end()) { in PrepareBackprop()347 CHECK(result.op_tape.emplace(op_id, op_it->second).second); in PrepareBackprop()360 op_tape->erase(op_it); in PrepareBackprop()374 for (const auto& op_pair : *op_tape) { in PrepareBackprop()378 op_tape->clear(); in PrepareBackprop()[all …]