Searched refs:shape_keys (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/slim/python/slim/data/ |
| D | tfexample_decoder.py | 152 def __init__(self, tensor_key, shape_keys=None, shape=None, default_value=0): argument
177 if shape_keys and shape is not None:
179 if shape_keys and not isinstance(shape_keys, list):
180 shape_keys = [shape_keys]
182 self._shape_keys = shape_keys
186 if shape_keys:
187 keys.extend(shape_keys)
217 shape_keys=None, argument
239 super(LookupTensor, self).__init__(tensor_key, shape_keys, shape,
|
| D | tfexample_decoder_test.py | 436 tfexample_decoder.Tensor('image', shape_keys='image/shape'),
438 tfexample_decoder.Tensor('labels', shape_keys='labels/shape'),
483 tfexample_decoder.Tensor('image', shape_keys='image/shape'),
487 shape_keys=['labels/height', 'labels/width', 'labels/depth']),
|