Searched refs:FunctionCacheKey (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/function/polymorphism/ |
| D | function_cache_test.py | 161 key_2 = function_cache.FunctionCacheKey(MockSubtypeOf2(2),
167 key_3 = function_cache.FunctionCacheKey(MockSubtypeOf2(3),
178 key_a = function_cache.FunctionCacheKey(generic(1),
180 key_b = function_cache.FunctionCacheKey(generic(2),
182 key_c = function_cache.FunctionCacheKey(generic(1),
191 key_a = function_cache.FunctionCacheKey(MockSubtypeOf2(1),
193 key_b = function_cache.FunctionCacheKey(MockSubtypeOf2(2),
195 key_c = function_cache.FunctionCacheKey(MockSubtypeOf2(1),
204 key_a = function_cache.FunctionCacheKey(MockSupertypes2With3(1),
206 key_b = function_cache.FunctionCacheKey(MockSupertypes2With3(2),
[all …]
|
| D | function_cache.py | 122 class FunctionCacheKey(trace.TraceType): class
141 if not isinstance(other, FunctionCacheKey):
153 isinstance(other, FunctionCacheKey) and
167 return FunctionCacheKey(args_common, captures_common, self.call_context)
183 if not isinstance(other, FunctionCacheKey):
217 def lookup(self, key: FunctionCacheKey, use_function_subtyping: bool): argument
228 def delete(self, key: FunctionCacheKey): argument
238 def add(self, key: FunctionCacheKey, argument
253 def generalize(self, key: FunctionCacheKey) -> FunctionCacheKey: argument
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_context.py | 126 ) -> Tuple[function_cache.FunctionCacheKey, trace_type.WeakrefDeletionObserver]:
138 return function_cache.FunctionCacheKey(
|