Searched refs:tensor_usage_counts (Results 1 – 1 of 1) sorted by relevance
510 std::unordered_map<int64, int64> tensor_usage_counts; member549 auto count_it = result.tensor_usage_counts.find(it); in PrepareBackprop()550 if (count_it != result.tensor_usage_counts.end()) { in PrepareBackprop()553 result.tensor_usage_counts[it] = 1; in PrepareBackprop()563 for (auto& pair : result.tensor_usage_counts) { in PrepareBackprop()824 auto usage_count_it = state.tensor_usage_counts.find(id); in ComputeGradient()825 if (usage_count_it == state.tensor_usage_counts.end()) { in ComputeGradient()