Searched refs:src_node_shape_attr (Results 1 – 1 of 1) sorted by relevance
513 const auto* src_node_shape_attr = src_node->GetAttr(kAttrOutputShape); in UpdateEdge() local514 if (src_node_shape_attr != nullptr) { in UpdateEdge()515 input_shape_proto = src_node_shape_attr->list().shape(src_port); in UpdateEdge()