Searched refs:op_missing_tensor (Results 1 – 1 of 1) sorted by relevance
314 gtl::FlatMap<int64, int64> op_missing_tensor; member366 result.op_missing_tensor[it->second] += 1; in PrepareBackprop()386 const gtl::FlatMap<int64, int64>& op_missing_tensor) { in InitialStack() argument389 if (op_missing_tensor.find(op_entry.first) == op_missing_tensor.end()) { in InitialStack()487 InitialStack(state.op_tape, state.op_missing_tensor); in ComputeGradient()631 auto missing_it = state.op_missing_tensor.find(op_id); in ComputeGradient()632 if (missing_it != state.op_missing_tensor.end()) { in ComputeGradient()