Searched refs:available_tensors_ (Results 1 – 1 of 1) sorted by relevance
1107 auto available_it = available_tensors_.find(batch_index(i, 0)); in OutputBatch()1108 if (available_it == available_tensors_.end()) { in OutputBatch()1112 available_tensors_.erase(available_it); in OutputBatch()1143 if (!available_tensors_.emplace(batch_key, grad_t).second) { in Compute()1159 if (available_tensors_.find(batch_key) == available_tensors_.end()) { in Compute()1232 std::unordered_map<int64, Tensor> available_tensors_; member in tensorflow::UnbatchGradResource