Home
last modified time | relevance | path

Searched refs:AddFunctionLibrary (Results 1 – 25 of 38) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dbuild_xla_ops_pass_test.cc138 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
160 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
180 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
216 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
245 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
284 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
H A Dresource_operation_safety_analysis_test.cc265 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
287 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
307 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
327 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
349 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
376 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
H A Dcompilability_check_util_test.cc322 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib)); in TEST_F()
419 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib)); in TEST_F()
474 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
502 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
551 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
H A Dencapsulate_xla_computations_pass_test.cc40 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto())); in MakeOuterGraph()
304 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST()
H A Dforce_xla_constants_on_host_pass_test.cc79 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_test.cc594 TEST_F(GraphTest, AddFunctionLibrary) { in TEST_F() argument
599 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
604 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
612 Status s = graph_.AddFunctionLibrary(error_proto); in TEST_F()
621 s = graph_.AddFunctionLibrary(error_proto); in TEST_F()
631 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
635 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
641 s = graph_.AddFunctionLibrary(error_proto); in TEST_F()
H A Dgraph_def_builder.h172 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() function
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc350 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
417 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
470 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
552 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
592 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
659 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
926 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
1009 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto())); in MakeOuterGraph()
1132 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST()
H A Dtfrt_graph_execution_state.cc823 optimized_graph->AddFunctionLibrary(optimized_flib->ToProto())); in OptimizeFunctions()
865 TF_RETURN_IF_ERROR(optimized_graph->AddFunctionLibrary(optimized_flib_proto)); in OptimizeGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op_test.cc66 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
182 TF_ASSERT_OK(graph->AddFunctionLibrary(f_lib_proto)); in TEST()
275 TF_ASSERT_OK(graph->AddFunctionLibrary(f_lib_proto)); in TEST()
415 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
531 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
H A Dlower_if_op_test.cc74 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
188 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
277 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
H A Dlower_function_call_op_test.cc91 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
155 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
H A Dlower_case_op_test.cc77 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
224 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
346 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
H A Dlower_functional_ops_test.cc150 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
258 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
H A Dfunction_test.cc816 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
832 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
933 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
945 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
969 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
1036 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
1117 TF_RETURN_IF_ERROR(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
1202 TF_RETURN_IF_ERROR(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
H A Dshape_refiner_test.cc1437 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TestSimpleFunctionInference()
1477 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1506 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1538 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
H A Dreplicate_per_replica_nodes_test.cc246 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties_test.cc1475 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib_)); in TEST_F()
1500 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib_)); in TEST_F()
1531 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1568 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1627 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1688 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
2407 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
2444 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
2490 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dexport_graphdef.cc527 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(*flib)); in Convert()
707 TF_RETURN_IF_ERROR(graph->get()->AddFunctionLibrary(flib)); in Convert()
753 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(flib)); in ConvertMlirToGraphdef()
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dfunctional_grad_test.cc67 TF_ASSERT_OK(scope_.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dwhile_op_test.cc187 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dgraph_compiler_util.cc306 TF_RETURN_IF_ERROR(g->AddFunctionLibrary(flib_def.ToProto())); in InitGraph()
H A Dfunctionalize_control_flow.cc126 return graph->AddFunctionLibrary(fdef_lib); in AddFunctionDefToGraphLibrary()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_constants_lib.cc217 TF_RETURN_IF_ERROR(input_graph.AddFunctionLibrary(input_graph_def.library())); in FoldConstants()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_function.cc264 status->status = g->graph.AddFunctionLibrary(fdef_lib); in TF_GraphCopyFunction()

12