Searched refs:TFE_TensorHandleCache (Results 1 – 5 of 5) sorted by relevance
33 TFE_TensorHandleCache* TFE_TensorHandleCache::Get() { in Get()36 static auto* cache = new TFE_TensorHandleCache(); in Get()40 TFE_TensorHandle* TFE_TensorHandleCache::Lookup( in Lookup()55 void TFE_TensorHandleCache::Insert(PyObject* value, tensorflow::DataType dtype, in Insert()64 void TFE_TensorHandleCache::Clear() { in Clear()
69 struct TFE_TensorHandleCache { struct70 static TFE_TensorHandleCache* Get();72 TFE_TensorHandleCache() { cache.reserve(64); } in TFE_TensorHandleCache() function73 ~TFE_TensorHandleCache() { DecrefUnrefAll(); } in ~TFE_TensorHandleCache() argument
339 auto* cache = TFE_TensorHandleCache::Get(); in ConvertToEagerTensor()
143 tensorflow::TFE_TensorHandleCache144 tensorflow::TFE_TensorHandleCache::Clear
296 tensorflow::TFE_TensorHandleCache::Get()->Clear(); in InputTFE_OutputTensorHandles()