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

use outputs directly

Browse Source
This commit is contained in:
Võ Văn Nghĩa 2020-12-15 13:18:38 +07:00
parent 9aef3f6fd1
commit 462a06442a
1 changed files with 2 additions and 3 deletions

5
tensorflow/c/eager/gradient_checker.cc
View File

@ -82,9 +82,8 @@ Status RunAndMaybeSum(AbstractContext* ctx, Model forward,
}
// Reduce sum the output on all dimensions.
TF_RETURN_IF_ERROR(ops::Sum(ctx, {model_out.get(), sum_dims.get()},
absl::MakeSpan(model_outputs), "sum_output"));
outputs[0] = model_outputs[0];
TF_RETURN_IF_ERROR(
ops::Sum(ctx, {model_out.get(), sum_dims.get()}, outputs, "sum_output"));
return Status::OK();
}
// ========================= End Helper Functions==============================