Searched refs:op_missing_tensor (Results 1 – 1 of 1) sorted by relevance
514 std::unordered_map<int64, int64> op_missing_tensor; member566 result.op_missing_tensor[it->second] += 1; in PrepareBackprop()586 const std::unordered_map<int64, int64>& op_missing_tensor) { in InitialStack() argument589 if (op_missing_tensor.find(op_entry.first) == op_missing_tensor.end()) { in InitialStack()694 InitialStack(state.op_tape, state.op_missing_tensor); in ComputeGradient()852 auto missing_it = state.op_missing_tensor.find(op_id); in ComputeGradient()853 if (missing_it != state.op_missing_tensor.end()) { in ComputeGradient()