Searched defs:TensorInfoCache (Results 1 – 1 of 1) sorted by relevance
789 struct TensorInfoCache {}; struct in tensorflow::SparseMatMul873 struct TensorInfoCache { struct in tensorflow::LibxsmmSparseMatMul874 struct TensorInfoCacheEntry {889 tensorflow::mutex lock;898 TensorInfoCache() : lock(), entries() {} in TensorInfoCache() function901 std::unique_ptr<TensorInfoCacheEntry> take_cache_entry(int M, int K, int N, in take_cache_entry()921 void return_cache_entry(std::unique_ptr<TensorInfoCacheEntry> e) in return_cache_entry()927 ~TensorInfoCache() { in ~TensorInfoCache()1531 typename SparseMatMul<TL, TR>::TensorInfoCache* /*cache*/, in Compute()