Searched refs:desired_tensor_to_batch_map_ (Results 1 – 1 of 1) sorted by relevance
1173 if (!desired_tensor_to_batch_map_.emplace(i, batch_key).second) { in Compute()1189 auto desire_it = desired_tensor_to_batch_map_.find(batch_key); in Compute()1190 if (desire_it != desired_tensor_to_batch_map_.end()) { in Compute()1193 desired_tensor_to_batch_map_.erase(desire_it); in Compute()1236 std::unordered_map<int64, int64> desired_tensor_to_batch_map_; member in tensorflow::UnbatchGradResource