Home
last modified time | relevance | path

Searched defs:func_graph (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcond_v2.py390 def _grad_fn(func_graph, grads): argument
428 def _create_grad_func(func_graph, grads, name): argument
488 def _get_intermediates(func_graph): argument
764 def _wrap_intermediates(func_graph, intermediates): argument
769 def _create_dummy_input(func_graph, template_tensor): argument
784 def _create_none_optionals(func_graph, n): argument
798 def _create_fakeparams(func_graph, template_tensors): argument
H A Dwhile_v2.py511 def _get_intermediates(func_graph): argument
727 def _grad_fn(ys, xs, args, func_graph): argument
H A Dcontrol_flow_util_v2.py60 def create_new_tf_function(func_graph): argument
H A Dgradients_util.py376 def _Captures(func_graph): argument
H A Dfunctional_ops.py1232 def _set_read_only_resource_inputs_attr(op, func_graph): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction.py556 def __init__(self, func_graph, attrs, func_graph_deleter): argument
783 def __init__(self, func_graph, attrs, func_graph_deleter, argument
1226 def __init__(self, func_graph, attrs, func_graph_deleter, argument
1396 def __init__(self, func_graph, attrs=None, shared_func_graph=True, spec=None): argument
3246 def __init__(self, func_graph): argument
H A Ddef_function.py515 def __init__(self, func_graph): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dfunc_graph.py1090 def func_graph_from_py_func(name, argument
1539 def dismantle_func_graph(func_graph): argument
1554 def override_func_graph_name_scope(func_graph, name_scope): argument
H A Dfunction_def_to_graph.py304 def _set_handle_data(func_graph, fdef): argument
H A Dauto_control_deps_utils.py32 def get_read_only_resource_input_indices_graph(func_graph): argument
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_function_test.cc1216 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in DefineFunction() local
1250 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in TEST_F() local
1292 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in TEST_F() local
1594 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in DefineStatefulFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dfunction_deserialization.py476 def _restore_gradient_functions(func_graph, renamed_functions, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util.cc247 Graph* func_graph = fbody->graph; in PropagateConstIntoFuncAttr() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfunction_extraction.cpp474 auto func_graph = ConstructFuncGraph(func_ctx); in CreateFunctionDefNode() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dfunction.cc415 FunctionBody* func_graph = nullptr; member