Searched refs:shape_node1 (Results 1 – 1 of 1) sorted by relevance
604 const NodeDef* shape_node1 = node_map_->GetNode(node.input(0)); in MaterializeBroadcastGradientArgs() local606 if (shape_node1 == nullptr || in MaterializeBroadcastGradientArgs()607 (shape_node1->op() != "Shape" && !IsReallyConstant(*shape_node1)) || in MaterializeBroadcastGradientArgs()620 if (!ExtractShape(*shape_node1, properties, &shape1, &min_id)) { in MaterializeBroadcastGradientArgs()