Searched refs:desired_tensor_to_batch_map_ (Results 1 – 1 of 1) sorted by relevance
843 if (!desired_tensor_to_batch_map_.emplace(i, batch_key).second) { in Compute()859 auto desire_it = desired_tensor_to_batch_map_.find(batch_key); in Compute()860 if (desire_it != desired_tensor_to_batch_map_.end()) { in Compute()863 desired_tensor_to_batch_map_.erase(desire_it); in Compute()906 std::unordered_map<int64, int64> desired_tensor_to_batch_map_; member in tensorflow::UnbatchGradResource