Home
last modified time | relevance | path

Searched refs:function_context (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/function/polymorphism/
H A Dfunction_cache_test.py23 from tensorflow.python.eager import function_context
120 key_1, deletion_observer_1 = function_context.make_cache_key(1, {})
123 key_2, deletion_observer_2 = function_context.make_cache_key(2, {})
124 key_3, _ = function_context.make_cache_key(3, {})
136 key_1, deletion_observer_1 = function_context.make_cache_key(1, {})
137 key_2, deletion_observer_2 = function_context.make_cache_key(2, {})
138 key_3, _ = function_context.make_cache_key(3, {})
155 key_1, deletion_observer_1 = function_context.make_cache_key(1)
308 key, deletion_observer = function_context.make_cache_key(dummy_object)
323 key, deletion_observer = function_context.make_cache_key(
[all …]
H A DBUILD56 "//tensorflow/python/eager:function_context",
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/converters/
H A Dfunctions.py71 function_context=function_context_name,
119 function_context=function_context_name,
H A Dreturn_statements.py372 function_context=anno.getanno(node, 'function_context_name'),
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
H A Dc_api_unified_experimental_mlir.cc133 MlirFunctionContext* function_context) in MlirAbstractOp() argument
136 function_context_(function_context) {} in MlirAbstractOp()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A DBUILD629 ":function_context",
650 name = "function_context",
651 srcs = ["function_context.py"],
H A Dfunction.py35 from tensorflow.python.eager import function_context
2668 lookup_func_key, _ = function_context.make_cache_key((args, kwargs),
2702 function_context.make_cache_key((args, kwargs), captures))