Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc139 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
161 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
181 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
217 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
246 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
285 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
Dcompilability_check_util_test.cc292 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib)); in TEST_F()
389 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib)); in TEST_F()
444 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
472 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
521 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
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()
Dforce_xla_constants_on_host_pass_test.cc79 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST()
Dencapsulate_xla_computations_pass_test.cc39 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto())); in MakeOuterGraph()
310 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.h168 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() function
169 return graph_.AddFunctionLibrary(fdef_lib); in AddFunctionLibrary()
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()
Dgraph.h583 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib);
/external/tensorflow/tensorflow/core/common_runtime/
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()
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()
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()
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()
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()
Dpartitioning_utils.cc64 TF_CHECK_OK(subgraph->AddFunctionLibrary(global_flib.ToProto())); in PartitionFunctionGraph()
Dfunction_test.cc823 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
839 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
940 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
952 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
976 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
1043 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
1124 TF_RETURN_IF_ERROR(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
1209 TF_RETURN_IF_ERROR(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
Dreplicate_per_replica_nodes_test.cc250 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST()
Dshape_refiner_test.cc1389 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TestSimpleFunctionInference()
1429 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1458 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1490 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1305 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib_)); in TEST_F()
1330 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib_)); in TEST_F()
1361 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1398 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1457 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1518 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
2237 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); in TEST_F()
2274 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
2320 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(function_lib)); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc306 TF_RETURN_IF_ERROR(g->AddFunctionLibrary(flib_def.ToProto())); in InitGraph()
Dfunctionalize_control_flow.cc127 return graph->AddFunctionLibrary(fdef_lib); in AddFunctionDefToGraphLibrary()
Dfunctionalize_control_flow_test.cc228 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in RunTest()
450 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(fdef_lib)); in AddNoinlineFunctionToGraph()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc217 TF_RETURN_IF_ERROR(input_graph.AddFunctionLibrary(input_graph_def.library())); in FoldConstants()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc263 status->status = g->graph.AddFunctionLibrary(fdef_lib); in TF_GraphCopyFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc484 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(*flib)); in Convert()
554 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(*flib)); in Convert()
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc293 TF_RETURN_IF_ERROR(b_->AddFunctionLibrary(def)); in AddFunction()

12