Searched refs:shape_inference_graph_interim (Results 1 – 1 of 1) sorted by relevance
1914 auto shape_inference_graph_interim = in InferShapesWithResourceVar() local1916 CopyGraph(*graph, shape_inference_graph_interim.get()); in InferShapesWithResourceVar()1918 for (Node* node : shape_inference_graph_interim->nodes()) { in InferShapesWithResourceVar()1933 [variable_edge, graph = shape_inference_graph_interim.get(), node] { in InferShapesWithResourceVar()1941 [graph = shape_inference_graph_interim.get(), out_edge, read_node] { in InferShapesWithResourceVar()1975 shape_inference_graph_interim.get(), arg_shapes, in InferShapesWithResourceVar()1976 &shape_inference_graph_interim->flib_def(), tpu_inferred_info)); in InferShapesWithResourceVar()