diff options
Diffstat (limited to 'tensorflow/python/util/util.h')
| -rw-r--r-- | tensorflow/python/util/util.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/util/util.h b/tensorflow/python/util/util.h index 70efc10c9a..41dcc969f8 100644 --- a/tensorflow/python/util/util.h +++ b/tensorflow/python/util/util.h @@ -118,7 +118,9 @@ PyObject* Flatten(PyObject* nested); // the type from the module. This approach also requires some trigger from // Python so that we know that Python interpreter had been initialzied. void RegisterSequenceClass(PyObject* sequence_class); -// Similar to the above function, except for the +// Like RegisterSequenceClass, but for collections.Mapping. +void RegisterMappingClass(PyObject* mapping_class); +// Similar to the above functions, except for the // sparse_tensor.SparseTensorValue class. void RegisterSparseTensorValueClass(PyObject* sparse_tensor_value_class); |