コード例 #1
0
ファイル: nn_ops.py プロジェクト: ThomasWollmann/tensorflow 
def _MaxPoolWithArgMaxShape(op):
  """Shape function for MaxPoolWithArgmax op."""
  return common_shapes.max_pool_shape(op) * 2
コード例 #2
0
ファイル: nn_ops.py プロジェクト: zzhang1987/tensorflow 
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()]