Searched defs:shape_node (Results 1 – 10 of 10) sorted by relevance
76 Status TryToInferTensorOutputFromShapeNode(const Node& shape_node, in TryToInferTensorOutputFromShapeNode()136 const Node* shape_node = edge->src(); in TryToInferTensorOutputFromStridedSliceNode() local
159 Node* shape_node; in RewriteIntoTempVariable() local
116 auto shape_node = originalInputs.at(0)->node(); in ImplicitCastForBinaryInplaceOps() local
86 auto* shape_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
206 Node* shape_node = nullptr; in ReplaceBroadcast() local
455 auto* shape_node = graph_view.GetNode("shape"); in TEST_F() local
566 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()2928 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
1753 auto shape_node = n->input(0)->node(); in SpecialPostProcess() local
1535 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local