Searched defs:TensorInfoCache (Results 1 – 1 of 1) sorted by relevance
788 struct TensorInfoCache {}; struct in tensorflow::SparseMatMul872 struct TensorInfoCache { struct in tensorflow::LibxsmmSparseMatMul873 struct TensorInfoCacheEntry {888 tensorflow::mutex lock;897 TensorInfoCache() : lock(), entries() {} in TensorInfoCache() function900 std::unique_ptr<TensorInfoCacheEntry> take_cache_entry(int M, int K, int N, in take_cache_entry()920 void return_cache_entry(std::unique_ptr<TensorInfoCacheEntry> e) in return_cache_entry()926 ~TensorInfoCache() { in ~TensorInfoCache()1528 typename SparseMatMul<TL, TR>::TensorInfoCache* /*cache*/, in Compute()