Searched refs:RewriteShapeInferenceGraph (Results 1 – 1 of 1) sorted by relevance
1202 Status RewriteShapeInferenceGraph(const string& shape_inference_graph_name, in RewriteShapeInferenceGraph() function2463 RewriteShapeInferenceGraph(shape_inference_graph, host_graph.get(), in ExtractOutsideCompilationForFunction()2537 TF_RETURN_IF_ERROR(RewriteShapeInferenceGraph( in ExtractOutsideCompilation()