[shape inference] Fix dim type for Cast

ChunliF · facebook-github-bot · commit 7929bc76a059 · 2021-04-21T03:21:56.000-07:00
Summary: ATT

Test Plan: unit test

Reviewed By: yinghai

Differential Revision: D27904584

fbshipit-source-id: b62d2eb5da0be79091c82e6300dd0c075a0bf2fe
diff --git a/caffe2/opt/bound_shape_inferencer.cc b/caffe2/opt/bound_shape_inferencer.cc
@@ -960,7 +960,7 @@ void BoundShapeInferencer::InferTranspose(const OperatorDef& op) {
 
   auto it = shape_info_.find(op.input(0));
   if (it == shape_info_.end()) {
-    LOG(WARNING) << "Didn't find shape info for the input of Softmax";
+    LOG(WARNING) << "Didn't find shape info for the input of Transpose";
     return;
   }
 

Original file line number	Diff line number	Diff line change
`@@ -960,7 +960,7 @@ void BoundShapeInferencer::InferTranspose(const OperatorDef& op) {`
`960`	`960`
`961`	`961`	`auto it = shape_info_.find(op.input(0));`
`962`	`962`	`if (it == shape_info_.end()) {`
`963`		`- LOG(WARNING) << "Didn't find shape info for the input of Softmax";`
	`963`	`+ LOG(WARNING) << "Didn't find shape info for the input of Transpose";`
`964`	`964`	`return;`
`965`	`965`	`}`
`966`	`966`