diff options
Diffstat (limited to 'tensorflow/contrib/tensor_forest/kernels/stats_ops.cc')
| -rw-r--r-- | tensorflow/contrib/tensor_forest/kernels/stats_ops.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensor_forest/kernels/stats_ops.cc b/tensorflow/contrib/tensor_forest/kernels/stats_ops.cc index f80a34ece6..fe2c91c104 100644 --- a/tensorflow/contrib/tensor_forest/kernels/stats_ops.cc +++ b/tensorflow/contrib/tensor_forest/kernels/stats_ops.cc @@ -246,7 +246,8 @@ class ProcessInputOp : public OpKernel { const Tensor& input_weights = context->input(7); const Tensor& leaf_ids_tensor = context->input(8); - std::unique_ptr<TensorDataSet> data_set(new TensorDataSet(input_spec_, 0)); + std::unique_ptr<TensorDataSet> data_set( + new TensorDataSet(input_spec_, random_seed_)); data_set->set_input_tensors(input_data, sparse_input_indices, sparse_input_values, sparse_input_shape); |