예제 #1
0
def _MaxPoolWithArgMaxShape(op):
  """Shape function for MaxPoolWithArgmax op."""
  return common_shapes.max_pool_shape(op) * 2
예제 #2
0
def _QuantizedMaxPoolShape(op):
    return [
        common_shapes.max_pool_shape(op)[0],
        tensor_shape.scalar(),
        tensor_shape.scalar()
    ]
예제 #3
0
def _MaxPoolWithArgMaxShape(op):
    """Shape function for MaxPoolWithArgmax op."""
    return common_shapes.max_pool_shape(op) * 2
예제 #4
0
파일: nn_ops.py 프로젝트: 01bui/tensorflow 
def _QuantizedMaxPoolShape(op):
  return [common_shapes.max_pool_shape(op)[0], tensor_shape.scalar(),
          tensor_shape.scalar()]