Searched refs:shape_inference_graph_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 1202 Status RewriteShapeInferenceGraph(const string& shape_inference_graph_name, in RewriteShapeInferenceGraph() argument
1212 fld->Find(shape_inference_graph_name); in RewriteShapeInferenceGraph()
1228 shape_inference_graph_name, in RewriteShapeInferenceGraph()
1310 DumpGraphToFile(shape_inference_graph_name, *g, fld); in RewriteShapeInferenceGraph()
1316 GraphToFunctionDef(*g, shape_inference_graph_name, &fdef_replace)); in RewriteShapeInferenceGraph()
1318 fld->ReplaceFunction(shape_inference_graph_name, fdef_replace)); in RewriteShapeInferenceGraph()
2536 for (const auto& shape_inference_graph_name : shape_inference_graphs) { in ExtractOutsideCompilation() local
2538 shape_inference_graph_name, g, pivot_node, fld)); in ExtractOutsideCompilation()
|