Searched defs:shape_node (Results 1 – 5 of 5) sorted by relevance
127 Node* shape_node; in rewriteNode() local
206 Node* shape_node = nullptr; in ReplaceBroadcast() local
447 Node* shape_node; in StackTensor() local
508 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()2583 NodeDef* shape_node = optimized_graph->add_node(); in SimplifyReduction() local
1388 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local