Searched refs:tensor_usage_counts (Results 1 – 1 of 1) sorted by relevance
310 gtl::FlatMap<int64, int64> tensor_usage_counts; member349 auto count_it = result.tensor_usage_counts.find(it); in PrepareBackprop()350 if (count_it != result.tensor_usage_counts.end()) { in PrepareBackprop()353 result.tensor_usage_counts[it] = 1; in PrepareBackprop()363 for (auto& pair : result.tensor_usage_counts) { in PrepareBackprop()603 auto usage_count_it = state.tensor_usage_counts.find(id); in ComputeGradient()604 if (usage_count_it == state.tensor_usage_counts.end()) { in ComputeGradient()