Searched defs:shape_node (Results 1 – 9 of 9) sorted by relevance
86 auto* shape_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
159 Node* shape_node; in RewriteIntoTempVariable() local
206 Node* shape_node = nullptr; in ReplaceBroadcast() local
440 Node* shape_node; in StackTensor() local
485 auto* shape_node = graph_view.GetNode("shape"); in TEST_F() local
562 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()2878 NodeDef* shape_node = optimized_graph->add_node(); in SimplifyReduction() local
3092 auto* shape_node = context.graph_view->GetNode("shape"); in TEST_F() local3161 auto* shape_node = context.graph_view->GetNode("shape"); in TEST_F() local
1456 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local
2291 Node* shape_node = graph->AddNode(def, &status); in BuildDynamicShapeNodes() local