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

Update build_xla_ops_pass.cc

Browse Source
This commit is contained in:
tg-at-google 2020-07-29 17:27:29 -04:00 committed by GitHub
parent 181df6f3fc
commit 497525e2c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/jit/build_xla_ops_pass.cc
View File

@ -452,7 +452,7 @@ Status PredicateInt32Inputs(const Scope& root, Node* n,
root.graph()->AddControlEdge(predicate_as_control.node(), root.graph()->AddControlEdge(predicate_as_control.node(),
identity_n.operation.node()); identity_n.operation.node());
for (int32 i = 0, end = int32_inputs.size(); i < end; i++) { for (int i = 0, end = int32_inputs.size(); i < end; i++) {
TF_RETURN_IF_ERROR(root.graph()->UpdateEdge(identity_n[i].node(), i, n, TF_RETURN_IF_ERROR(root.graph()->UpdateEdge(identity_n[i].node(), i, n,
int32_inputs_input_idxs[i])); int32_inputs_input_idxs[i]));
} }