OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Pass stats_collector when using SymbolicGradientOp.

Browse Source 
PiperOrigin-RevId: 163773897
This commit is contained in:
A. Unique TensorFlower 2017-07-31 17:33:11 -07:00 committed by TensorFlower Gardener
parent 5202a5b6cf
commit 14b7367613
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/function_ops.cc
View File

@ -240,6 +240,7 @@ class SymbolicGradientOp : public AsyncOpKernel {
opts.rendezvous = ctx->rendezvous(); opts.rendezvous = ctx->rendezvous();
opts.cancellation_manager = ctx->cancellation_manager(); opts.cancellation_manager = ctx->cancellation_manager();
opts.runner = ctx->runner(); opts.runner = ctx->runner();
opts.stats_collector = ctx->stats_collector();
std::vector<Tensor> args; std::vector<Tensor> args;
args.reserve(ctx->num_inputs()); args.reserve(ctx->num_inputs());
for (int i = 0; i < ctx->num_inputs(); ++i) { for (int i = 0; i < ctx->num_inputs(); ++i) {