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