Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc138 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
158 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
178 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
212 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
239 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
Dresource_operation_safety_analysis_test.cc266 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
288 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
308 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
328 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
350 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
377 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
Dencapsulate_xla_computations_pass_test.cc38 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto())); in MakeOuterGraph()
311 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST()
Dmark_for_compilation_pass_test.cc831 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc593 TEST_F(GraphTest, AddFunctionLibrary) { in TEST_F() argument
598 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
603 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
611 Status s = graph_.AddFunctionLibrary(error_proto); in TEST_F()
620 s = graph_.AddFunctionLibrary(error_proto); in TEST_F()
630 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
634 TF_EXPECT_OK(graph_.AddFunctionLibrary(proto)); in TEST_F()
640 s = graph_.AddFunctionLibrary(error_proto); in TEST_F()
Dgraph_def_builder.h168 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() function
169 return graph_.AddFunctionLibrary(fdef_lib); in AddFunctionLibrary()
Dgraph.h532 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib);
Dgraph.cc602 Status Graph::AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() function in tensorflow::Graph
Dgraph_partition_test.cc465 TF_ASSERT_OK(in_.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
Dgraph_constructor.cc914 TF_RETURN_IF_ERROR(g_->AddFunctionLibrary(*library_)); in Convert()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op_test.cc66 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
179 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
268 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
Dlower_while_op_test.cc56 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
157 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
261 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
Dlower_if_while_test.cc138 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
246 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST()
Dpartitioning_utils.cc61 TF_CHECK_OK(subgraph->AddFunctionLibrary(global_flib.ToProto())); in PartitionFunctionGraph()
Dfunction_test.cc693 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
709 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
810 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
822 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
846 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
917 TF_ASSERT_OK(s.graph()->AddFunctionLibrary(fdef_lib_)); in TEST_F()
Dshape_refiner_test.cc1405 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TestSimpleFunctionInference()
1461 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1492 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1528 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
1560 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(f_lib_proto)); in TEST_F()
Dconstant_folding_test.cc345 TF_EXPECT_OK(s.graph()->AddFunctionLibrary(flib)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1125 TF_CHECK_OK(s.graph()->AddFunctionLibrary(function_lib_)); in TEST_F()
1150 TF_CHECK_OK(s.graph()->AddFunctionLibrary(function_lib_)); in TEST_F()
1181 TF_CHECK_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1218 TF_CHECK_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
1276 TF_CHECK_OK(s.graph()->AddFunctionLibrary(library)); in TEST_F()
/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/compiler/tf2xla/
Dtf2xla.cc373 TF_RETURN_IF_ERROR(g->AddFunctionLibrary(flib_def.ToProto())); in InitGraph()
Dxla_compiler_test.cc570 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
906 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST_F()
1349 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST_F()
Dfunctionalize_control_flow_test.cc342 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(fdef_lib)); in AddNoinlineFunctionToGraph()
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc273 TF_RETURN_IF_ERROR(b_->AddFunctionLibrary(def)); in AddFunction()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc665 status->status = g->graph.AddFunctionLibrary(fdef_lib); in TF_GraphCopyFunction()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc726 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(fdeflib)); in RegisterSegmentFunctionToFunctionLibrary()