Searched defs:TensorInfoCache (Results 1 – 1 of 1) sorted by relevance
784 struct TensorInfoCache {}; struct in tensorflow::SparseMatMul868 struct TensorInfoCache { struct in tensorflow::LibxsmmSparseMatMul869 struct TensorInfoCacheEntry {884 tensorflow::mutex lock;893 TensorInfoCache() : lock(), entries() {} in TensorInfoCache() argument896 std::unique_ptr<TensorInfoCacheEntry> take_cache_entry(int M, int K, int N, in take_cache_entry()916 void return_cache_entry(std::unique_ptr<TensorInfoCacheEntry> e) in return_cache_entry()922 ~TensorInfoCache() { in ~TensorInfoCache()1537 typename SparseMatMul<TL, TR>::TensorInfoCache* /*cache*/, in Compute()