Home
last modified time | relevance | path

Searched refs:_PARTITION_SHAPE (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dinit_ops_v2.py35 _PARTITION_SHAPE = "partition_shape" variable
104 if kwarg not in [_PARTITION_SHAPE, _PARTITION_OFFSET]:
152 if _PARTITION_SHAPE in kwargs:
153 shape = kwargs[_PARTITION_SHAPE]
198 if _PARTITION_SHAPE in kwargs:
199 shape = kwargs[_PARTITION_SHAPE]
351 if _PARTITION_SHAPE in kwargs:
352 shape = kwargs[_PARTITION_SHAPE]
417 if _PARTITION_SHAPE in kwargs:
418 shape = kwargs[_PARTITION_SHAPE]
[all …]
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v2.py35 _PARTITION_SHAPE = 'partition_shape' variable
153 if _PARTITION_SHAPE in kwargs:
154 shape = kwargs[_PARTITION_SHAPE]
190 if _PARTITION_SHAPE in kwargs:
191 shape = kwargs[_PARTITION_SHAPE]
293 if _PARTITION_SHAPE in kwargs:
294 shape = kwargs[_PARTITION_SHAPE]
353 if _PARTITION_SHAPE in kwargs:
354 shape = kwargs[_PARTITION_SHAPE]
418 if _PARTITION_SHAPE in kwargs:
[all …]