Searched refs:TensorAndKey (Results 1 – 3 of 3) sorted by relevance
60 struct TensorAndKey { struct71 Status AddTensor(const std::string& name, const TensorAndKey& tk); argument86 std::unordered_map<string, TensorAndKey> tensors_ TF_GUARDED_BY(lock_);
60 Status TensorStore::AddTensor(const string& name, const TensorAndKey& tk) { in AddTensor()
52 TensorStore::TensorAndKey tk{val, id, requested_device()}; in Compute()