Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/tpu/
Dtpu.py519 maximum_shapes=None): argument
575 maximum_shapes=maximum_shapes)[1]
598 maximum_shapes = []
600 maximum_shapes.append(
648 padding = [0, maximum_shapes[idx][i] - input_shape_tensor[i]]
667 maximum_shapes=None): argument
795 if maximum_shapes:
801 nest.assert_same_structure(inputs[0], maximum_shapes, check_types=False)
804 flat_maximum_shapes = nest.flatten(maximum_shapes)
856 if maximum_shapes is None: