Searched refs:desired_tensor_to_batch_map_ (Results 1 – 1 of 1) sorted by relevance
940 if (!desired_tensor_to_batch_map_.emplace(i, batch_key).second) { in Compute()956 auto desire_it = desired_tensor_to_batch_map_.find(batch_key); in Compute()957 if (desire_it != desired_tensor_to_batch_map_.end()) { in Compute()960 desired_tensor_to_batch_map_.erase(desire_it); in Compute()1003 std::unordered_map<int64_t, int64_t> desired_tensor_to_batch_map_; member in tensorflow::UnbatchGradResource