Searched refs:shape_node2 (Results 1 – 1 of 1) sorted by relevance
605 const NodeDef* shape_node2 = node_map_->GetNode(node.input(1)); in MaterializeBroadcastGradientArgs() local608 shape_node2 == nullptr || in MaterializeBroadcastGradientArgs()609 (shape_node2->op() != "Shape" && !IsReallyConstant(*shape_node2))) { in MaterializeBroadcastGradientArgs()624 if (!ExtractShape(*shape_node2, properties, &shape2, &min_id)) { in MaterializeBroadcastGradientArgs()