Searched refs:shape_inference_graph_interim (Results 1 – 1 of 1) sorted by relevance
1999 auto shape_inference_graph_interim = in InferShapesWithResourceVar() local2001 CopyGraph(*graph, shape_inference_graph_interim.get()); in InferShapesWithResourceVar()2003 for (Node* node : shape_inference_graph_interim->nodes()) { in InferShapesWithResourceVar()2018 [variable_edge, graph = shape_inference_graph_interim.get(), node] { in InferShapesWithResourceVar()2026 [graph = shape_inference_graph_interim.get(), out_edge, read_node] { in InferShapesWithResourceVar()2060 shape_inference_graph_interim.get(), arg_shapes, in InferShapesWithResourceVar()2061 &shape_inference_graph_interim->flib_def(), tpu_inferred_info)); in InferShapesWithResourceVar()