/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | cond_v2.py | 390 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 D | while_v2.py | 511 def _get_intermediates(func_graph): argument 727 def _grad_fn(ys, xs, args, func_graph): argument
|
H A D | control_flow_util_v2.py | 60 def create_new_tf_function(func_graph): argument
|
H A D | gradients_util.py | 376 def _Captures(func_graph): argument
|
H A D | functional_ops.py | 1232 def _set_read_only_resource_inputs_attr(op, func_graph): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
H A D | function.py | 556 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 D | def_function.py | 515 def __init__(self, func_graph): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
H A D | func_graph.py | 1090 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 D | function_def_to_graph.py | 304 def _set_handle_data(func_graph, fdef): argument
|
H A D | auto_control_deps_utils.py | 32 def get_read_only_resource_input_indices_graph(func_graph): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/c/ |
H A D | c_api_function_test.cc | 1216 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 D | function_deserialization.py | 476 def _restore_gradient_functions(func_graph, renamed_functions, argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | tf2xla_util.cc | 247 Graph* func_graph = fbody->graph; in PropagateConstIntoFuncAttr() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
H A D | function_extraction.cpp | 474 auto func_graph = ConstructFuncGraph(func_ctx); in CreateFunctionDefNode() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | function.cc | 415 FunctionBody* func_graph = nullptr; member
|