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