Fix typo (#10379)
This commit is contained in:
parent
3039d7da20
commit
8dcf37b471
@ -435,7 +435,7 @@ Status GraphConstructor::MakeNode(const NodeDef& node_def, Node** node) {
|
|||||||
Status GraphConstructor::ValidateShape(Node* node) {
|
Status GraphConstructor::ValidateShape(Node* node) {
|
||||||
if (!opts_.importing) return Status::OK();
|
if (!opts_.importing) return Status::OK();
|
||||||
TF_RETURN_IF_ERROR(refiner_->AddNode(node));
|
TF_RETURN_IF_ERROR(refiner_->AddNode(node));
|
||||||
// For nodes with the _output_shapes atttribute, override the shape.
|
// For nodes with the _output_shapes attribute, override the shape.
|
||||||
std::vector<TensorShapeProto> shape_attrs;
|
std::vector<TensorShapeProto> shape_attrs;
|
||||||
const char* kAttrName = "_output_shapes";
|
const char* kAttrName = "_output_shapes";
|
||||||
if (!GetNodeAttr(node->attrs(), kAttrName, &shape_attrs).ok()) {
|
if (!GetNodeAttr(node->attrs(), kAttrName, &shape_attrs).ok()) {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user