Home
last modified time | relevance | path

Searched refs:function_cache (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/function/polymorphism/
H A Dfunction_cache_test.py22 from tensorflow.core.function.polymorphism import function_cache
109 class MockEmptyCaptureSnapshot(function_cache.CaptureSnapshot):
118 cache = function_cache.FunctionCache()
134 cache = function_cache.FunctionCache()
154 cache = function_cache.FunctionCache()
161 key_2 = function_cache.FunctionCacheKey(MockSubtypeOf2(2),
167 key_3 = function_cache.FunctionCacheKey(MockSubtypeOf2(3),
176 ctx = function_cache.FunctionContext(0)
178 key_a = function_cache.FunctionCacheKey(generic(1),
180 key_b = function_cache.FunctionCacheKey(generic(2),
[all …]
H A DBUILD32 name = "function_cache",
33 srcs = ["function_cache.py"],
53 ":function_cache",
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction_context.py20 from tensorflow.core.function.polymorphism import function_cache
40 def make_function_context() -> function_cache.FunctionContext:
101 return function_cache.FunctionContext(
126 ) -> Tuple[function_cache.FunctionCacheKey, trace_type.WeakrefDeletionObserver]:
135 captures_signature = function_cache.CaptureSnapshot(
138 return function_cache.FunctionCacheKey(
H A DBUILD634 "//tensorflow/core/function/polymorphism:function_cache",
656 "//tensorflow/core/function/polymorphism:function_cache",
H A Dfunction.py27 from tensorflow.core.function.polymorphism import function_cache
2447 self._function_cache = function_cache.FunctionCache()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A DBUILD63 "function_cache.cc",
68 "function_cache.h",
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils.cc158 std::shared_ptr<FunctionCache> function_cache, in GraphHasher() argument
164 function_cache_(function_cache), in GraphHasher()
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dsave_test.py942 function_cache = root.f._stateful_fn._list_all_concrete_functions()
945 self.assertLen(function_cache, 1)
946 self.assertEqual(function_cache[0].name.decode("utf-8"),