Home
last modified time | relevance | path

Searched refs:tensor_usage_counts (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dtape.h510 std::unordered_map<int64, int64> tensor_usage_counts; member
549 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()