Пример #1
0
def Main():
  """."""
  cc = cc_emitter.CCEmitter()
  common.GenerateHeader(cc, 'gemmlowp_meta_streams_arm_64', 'GEMMLOWP_NEON_64')

  cc.EmitNamespaceBegin('gemmlowp')
  cc.EmitNamespaceBegin('meta')
  cc.EmitNewline()

  streams_common.GenerateUInt8x8Streams(cc, neon_emitter_64.NeonEmitter64(), 8)

  cc.EmitNamespaceEnd()
  cc.EmitNamespaceEnd()
  cc.EmitNewline()

  common.GenerateFooter(cc, 'Meta gemm for arm64 requires: GEMMLOWP_NEON_64!')
Пример #2
0
def Main():
    """."""
    cc = cc_emitter.CCEmitter()
    common.GenerateHeader(cc, 'gemmlowp_meta_transform_kernels_arm_32',
                          'GEMMLOWP_NEON_32')

    cc.EmitNamespaceBegin('gemmlowp')
    cc.EmitNamespaceBegin('meta')
    cc.EmitNewline()

    transform_kernels_common.GenerateKernels(cc, neon_emitter.NeonEmitter(),
                                             [(16, x) for x in range(16)])

    cc.EmitNamespaceEnd()
    cc.EmitNamespaceEnd()
    cc.EmitNewline()

    common.GenerateFooter(cc,
                          'Meta gemm for arm32 requires: GEMMLOWP_NEON_32!')
Пример #3
0
def Main():
    """."""
    cc = cc_emitter.CCEmitter()
    common.GenerateHeader(cc, 'gemmlowp_meta_quantized_mul_kernels_arm_64',
                          'GEMMLOWP_NEON_64')

    cc.EmitNamespaceBegin('gemmlowp')
    cc.EmitNamespaceBegin('meta')
    cc.EmitNewline()

    shapes = [(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (1, 7), (1, 8),
              (2, 1), (2, 2), (2, 3), (2, 4), (3, 1), (3, 2), (3, 3)]

    quantized_mul_kernels_common.GenerateKernels(
        cc, neon_emitter_64.NeonEmitter64(), shapes)

    cc.EmitNamespaceEnd()
    cc.EmitNamespaceEnd()
    cc.EmitNewline()

    common.GenerateFooter(cc,
                          'Meta gemm for arm64 requires: GEMMLOWP_NEON_64!')