experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove the condition of truncate

Browse Source
This commit is contained in:
Tzu-Wei Sung 2021-02-01 11:58:51 -08:00
parent a8200b1818
commit d4c4df4bbb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
View File

@ -222,7 +222,7 @@ bool MatchCallArgs(CallOp first, CallOp second, ArgMatcherFn matcher) {
// Consider cast compatibility in case
// %cast = "tf.Cast"(%0) : (tensor<2xi64>) -> tensor<2xf32>
// is skipped.
if ((cast_op.SrcT() != cast_op.DstT()) || cast_op.Truncate()) {
if (cast_op.SrcT() != cast_op.DstT()) {
break;
}
value = cast_op.getOperand();