| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 286 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
414 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
462 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
551 TF_CHECK_OK(GraphToFunctionDef(*g, "true_fn", true_fn_fdef)); in TEST_F()
567 TF_CHECK_OK(GraphToFunctionDef(*g, "false_fn", false_fn_fdef)); in TEST_F()
585 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
711 TF_CHECK_OK(GraphToFunctionDef(*g, "cond_fn", cond_fn_fdef)); in TEST_F()
727 TF_CHECK_OK(GraphToFunctionDef(*g, "body_fn", body_fn_fdef)); in TEST_F()
744 TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef)); in TEST_F()
842 TF_CHECK_OK(GraphToFunctionDef(*g, "fn", true_fn_fdef)); in TEST_F()
[all …]
|
| D | extract_outside_compilation_pass.cc | 591 GraphToFunctionDef(host_graph, host_graph_func_name, &host_graph_fdef)); in ConstructHostGraph()
797 GraphToFunctionDef(*g, shape_inference_graph_name, &fdef_replace)); in RewriteShapeInferenceGraph()
868 TF_RETURN_IF_ERROR(GraphToFunctionDef(*g, func_name, &replace_fdef)); in ReplaceKeyPlaceholderWithArgNode()
946 TF_RETURN_IF_ERROR(GraphToFunctionDef(host_graph, host_graph_func_name, in BuildHostGraphForIfNode()
1014 GraphToFunctionDef(*g, loop_cond_func.name(), &replace_fdef)); in AddSendLoopPredToLoopCond()
1093 GraphToFunctionDef(*cond_graph, cond_host_func_name, &cond_replace_fdef)); in RewriteHostWhileLoopCond()
1155 GraphToFunctionDef(*body_graph, body_host_func_name, &body_replace_fdef)); in RewriteHostWhileLoopBody()
1216 TF_RETURN_IF_ERROR(GraphToFunctionDef(host_graph, host_graph_func_name, in BuildHostGraphForWhileNode()
1264 TF_RETURN_IF_ERROR(GraphToFunctionDef(host_graph, host_graph_func_name, in BuildHostGraphForFuncCallNode()
1694 GraphToFunctionDef(*graph_out, new_func_name, &updated_fdef)); in ExtractOutsideCompilationForFunction()
|
| D | encapsulate_xla_computations_pass_test.cc | 303 TF_ASSERT_OK(GraphToFunctionDef(*body_graph, "launch0", flib.add_function())); in TEST()
|
| D | encapsulate_subgraphs_pass.cc | 1124 TF_RETURN_IF_ERROR(GraphToFunctionDef(*graph_, name, &fdef)); in BuildFunctionDef()
1173 GraphToFunctionDef(*inference_graph, inference_graph_name, &fdef)); in AddShapeInferenceInfo()
1189 TF_RETURN_IF_ERROR(GraphToFunctionDef(*graph_, name, &fdef)); in ReplaceFunctionDef()
|
| D | encapsulate_subgraphs_pass_test.cc | 58 TF_RETURN_IF_ERROR(GraphToFunctionDef( in AddGraphDefToFunctionLibrary()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef_test.cc | 73 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef)); in TEST()
110 TF_EXPECT_OK(GraphToFunctionDef(*graph, "test_fn", &fdef)); in TEST()
|
| D | graph_to_functiondef.h | 27 Status GraphToFunctionDef(const Graph& graph, const string& name,
|
| D | graph_to_functiondef.cc | 116 Status GraphToFunctionDef(const Graph& graph, const string& name, in GraphToFunctionDef() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 345 TF_ASSERT_OK(GraphToFunctionDef(graph, "cond", &cond_fdef)); in TEST()
348 TF_ASSERT_OK(GraphToFunctionDef(graph, "body", &body_fdef)); in TEST()
378 TF_ASSERT_OK(GraphToFunctionDef(graph, "cond", &cond_fdef)); in TEST()
381 TF_ASSERT_OK(GraphToFunctionDef(graph, "body", &body_fdef)); in TEST()
|
| D | functionalize_control_flow.cc | 217 GraphToFunctionDef(*g, new_func_name, &functionalized_fdef)); in FunctionalizeControlFlowForFunction()
|
| D | functionalize_while.cc | 503 GraphToFunctionDef(*cond_graph, cond_name.name(), &cond_fdef)); in FunctionalizeLoop()
506 GraphToFunctionDef(*body_graph, body_name.name(), &body_fdef)); in FunctionalizeLoop()
|
| D | tf2xla_util.cc | 206 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef)); in PropagateConstIntoFuncAttr()
|
| D | functionalize_cond.cc | 743 TF_RETURN_IF_ERROR(GraphToFunctionDef(*bodies_[branch_index], in BuildIfNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 624 TF_RETURN_IF_ERROR(GraphToFunctionDef( in GetResult()
635 TF_RETURN_IF_ERROR(GraphToFunctionDef( in GetResult()
|
| D | vectorization_utils_test.cc | 91 return GraphToFunctionDef(graph, "outer_function", result); in WrapFunctionWithMapDefun()
912 TF_RETURN_IF_ERROR(GraphToFunctionDef(graph, "inner_function", &inner)); in CwiseTestHelper()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 299 Status GraphToFunctionDef(const Graph& fn_body, const string& fn_name, in GraphToFunctionDef() function
616 status->status = tensorflow::GraphToFunctionDef( in TF_GraphToFunctionWithControlOutputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 713 TF_RETURN_IF_ERROR(GraphToFunctionDef( in RegisterSegmentFunctionToFunctionLibrary()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 677 TF_RETURN_IF_ERROR(GraphToFunctionDef(*subgraph, unique_name, &shard)); in InstantiateMultiDevice()
|