Home
last modified time | relevance | path

Searched refs:enable_nested_function_shape_inference (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/config/
H A Dflag_defs.h32 TF_DECLARE_FLAG(enable_nested_function_shape_inference, false,
H A Dflags_api_wrapper.cc46 TF_PY_DECLARE_FLAG(enable_nested_function_shape_inference); in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtest_util.py1096 def enable_nested_function_shape_inference(fn): function
1121 if flags.config().enable_nested_function_shape_inference.value():
1124 flags.config().enable_nested_function_shape_inference.reset(True)
1128 flags.config().enable_nested_function_shape_inference.reset(False)