Home
last modified time | relevance | path

Searched refs:function_cache_ (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils.cc152 function_cache_ = std::make_shared<FunctionCache>(); in GraphHasher()
164 function_cache_(function_cache), in GraphHasher()
527 auto it = function_cache_->find(fdef); in HashFunction()
528 if (it != function_cache_->end()) { in HashFunction()
545 function_cache_, attr_cache_); in HashFunction()
561 auto result = function_cache_->emplace(fdef, *hash); in HashFunction()
615 function_cache_, attr_cache_); in CheckFunctionsEqualHelper()
618 node_cache_, function_cache_, attr_cache_); in CheckFunctionsEqualHelper()
716 std::shared_ptr<FunctionCache> function_cache_; member in tensorflow::data::__anon4acced740111::GraphHasher
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.h258 FunctionCache& GetFunctionCache() { return function_cache_; } in GetFunctionCache()
302 FunctionCache function_cache_; variable
H A Dc_api_tfrt.cc1107 function_cache_.RemoveFunction(func); in RemoveFunction()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc562 auto iter = function_cache_.find(cache_key); in GetCachedFunction()
565 if (iter != function_cache_.end()) { in GetCachedFunction()
617 return &function_cache_.insert({cache_key, std::move(function)}) in AddCachedFunction()
H A Ddtensor_device_util.h581 function_cache_; variable