frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[shape inference] Fix dim type for Cast

Browse Source 
Summary: ATT

Test Plan: unit test

Reviewed By: yinghai

Differential Revision: D27904584

fbshipit-source-id: b62d2eb5da0be79091c82e6300dd0c075a0bf2fe
This commit is contained in:
Chunli Fu
2021-04-21 03:20:40 -07:00
committed by Facebook GitHub Bot
parent 4575028f6c
commit 7929bc76a0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/opt/bound_shape_inferencer.cc
View File

@ -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;
}