Esempi in Python per OpsRegister

Esempio n. 1

File: ops.py Progetto: suntaiwu/Anakin

#! /usr/bin/env python
# Copyright (c) 2017, Cuichaowen. All rights reserved.
# -*- coding: utf-8 -*-
from op import OpsParam, OpsRegister
from op_io import *

############################# IO define ##############################
# graph may has mult-inputs, so graph will have multi-input
OpsRegister.Register("Input").set_attr(input_shape=list(),
                                       max_len=int(),
                                       max_batch=int(),
                                       alias="NULL",
                                       data_type="NULL")

# graph out , only hold place for edge
OpsRegister.Register("Output").set_attr()

OpsRegister.Register("Split").set_attr(split_num=int())

############################# Basic Op define ##############################
# two input
OpsRegister.Register("Dot").set_attr(axes=list())
# one or two input
# enum type {
#		 Add,
#		 Subtract,
#		 Multiply,
#		 Avg,
#		 Max
#	  }
#  note : coeff only used by caffe for "Add"

Esempio n. 2

File: ops_fluid.py Progetto: suntaiwu/Anakin

#! /usr/bin/env python
# Copyright (c) 2017, Cuichaowen. All rights reserved.
# -*- coding: utf-8 -*-
from op import OpsParam, OpsRegister
from op_io import *

OpsRegister.Register("elementwise_mul").set_attr()
OpsRegister.Register("depthwise_conv2d").set_attr()
OpsRegister.Register("transpose").set_attr()
OpsRegister.Register("reshape").set_attr()
OpsRegister.Register("concat").set_attr()
OpsRegister.Register("box_coder").set_attr()

OpsRegister.Register("im2sequence").set_attr()
OpsRegister.Register("sum").set_attr()
OpsRegister.Register("top_k").set_attr()
OpsRegister.Register("ctc_align").set_attr()
OpsRegister.Register("cast").set_attr()
OpsRegister.Register("elementwise_add_fulid").set_attr()

OpsRegister.Register("lookup_table").set_attr()
OpsRegister.Register("lstm").set_attr()
OpsRegister.Register("sequence_pool").set_attr()
OpsRegister.Register("tanh").set_attr()

OpsRegister.Register("sequence_conv").set_attr()
OpsRegister.Register("stanh").set_attr()

OpsRegister.Register("matmul").set_attr()
OpsRegister.Register("layer_norm").set_attr()
OpsRegister.Register("dropout").set_attr()

Esempio n. 3

#! /usr/bin/env python
# Copyright (c) 2017, Cuichaowen. All rights reserved.
# -*- coding: utf-8 -*-
from op import OpParam
from op import OpsRegister
from op_io import *

############################# IO define ##############################
# graph may has mult-inputs, so graph will have multi-input
OpsRegister.Register("Input").set_attr(input_shape=list())

# graph out , only hold place for edge
OpsRegister.Register("Output").set_attr()

OpsRegister.Register("Split").set_attr(split_num=int())

############################# Basic Op define ##############################
# two input
OpsRegister.Register("Dot").set_attr(axes=list())
# one or two input
# enum type {
#		 Add,
#		 Subtract,
#		 Multiply,
#		 Avg,
#		 Max
#	  }
#  note : coeff only used by caffe for "Add"
OpsRegister.Register("Eltwise").set_attr(type="Add", coeff=list())
# list input
OpsRegister.Register("Concat").set_attr(axis=int())

Esempio n. 4

File: ops_fluid.py Progetto: zhaoyang-star/Anakin

#! /usr/bin/env python
# Copyright (c) 2017, Cuichaowen. All rights reserved.
# -*- coding: utf-8 -*-
from op import OpsParam, OpsRegister
from op_io import *

OpsRegister.Register("elementwise_mul").set_attr()
OpsRegister.Register("depthwise_conv2d").set_attr()
OpsRegister.Register("transpose").set_attr()
OpsRegister.Register("reshape").set_attr()
OpsRegister.Register("concat").set_attr()
OpsRegister.Register("box_coder").set_attr()

OpsRegister.Register("im2sequence").set_attr()
OpsRegister.Register("sum").set_attr()
OpsRegister.Register("top_k").set_attr()
OpsRegister.Register("ctc_align").set_attr()
OpsRegister.Register("cast").set_attr()
OpsRegister.Register("elementwise_add_fulid").set_attr()

OpsRegister.Register("lookup_table").set_attr()
OpsRegister.Register("lstm").set_attr()
OpsRegister.Register("sequence_pool").set_attr()
OpsRegister.Register("tanh").set_attr()

OpsRegister.Register("sequence_conv").set_attr()
OpsRegister.Register("stanh").set_attr()

OpsRegister.Register("matmul").set_attr()
OpsRegister.Register("layer_norm").set_attr()
OpsRegister.Register("dropout").set_attr()

Esempio n. 5

#! /usr/bin/env python
# Copyright (c) 2017, Cuichaowen. All rights reserved.
# -*- coding: utf-8 -*-
from op import OpsParam, OpsRegister
from op_io import *

OpsRegister.Register("elementwise_mul").set_attr()
OpsRegister.Register("depthwise_conv2d").set_attr()
OpsRegister.Register("transpose").set_attr()
OpsRegister.Register("reshape").set_attr()
OpsRegister.Register("concat").set_attr()
OpsRegister.Register("box_coder").set_attr()

OpsRegister.Register("im2sequence").set_attr()
OpsRegister.Register("sum").set_attr()
OpsRegister.Register("top_k").set_attr()
OpsRegister.Register("ctc_align").set_attr()
OpsRegister.Register("cast").set_attr()
OpsRegister.Register("elementwise_add_fulid").set_attr()

OpsRegister.Register("lookup_table").set_attr()
OpsRegister.Register("lstm").set_attr()
OpsRegister.Register("sequence_pool").set_attr()
OpsRegister.Register("tanh").set_attr()

OpsRegister.Register("sequence_conv").set_attr()
OpsRegister.Register("stanh").set_attr()

OpsRegister.Register("matmul").set_attr()
OpsRegister.Register("layer_norm").set_attr()
OpsRegister.Register("dropout").set_attr()

Esempio n. 6

File: ops_fluid.py Progetto: zewu82/Anakin

#! /usr/bin/env python
# Copyright (c) 2017, Cuichaowen. All rights reserved.
# -*- coding: utf-8 -*-
from op import OpsParam, OpsRegister
from op_io import *

OpsRegister.Register("elementwise_mul").set_attr()
OpsRegister.Register("depthwise_conv2d").set_attr()
OpsRegister.Register("transpose").set_attr()
OpsRegister.Register("reshape").set_attr()
OpsRegister.Register("concat").set_attr()
OpsRegister.Register("box_coder").set_attr()

OpsRegister.Register("im2sequence").set_attr()
OpsRegister.Register("sum").set_attr()
OpsRegister.Register("top_k").set_attr()
OpsRegister.Register("ctc_align").set_attr()
OpsRegister.Register("cast").set_attr()
OpsRegister.Register("elementwise_add_fulid").set_attr()

OpsRegister.Register("lookup_table").set_attr()
OpsRegister.Register("lstm").set_attr()
OpsRegister.Register("sequence_pool").set_attr()
OpsRegister.Register("tanh").set_attr()

OpsRegister.Register("sequence_conv").set_attr()
OpsRegister.Register("stanh").set_attr()

OpsRegister.Register("matmul").set_attr()
OpsRegister.Register("layer_norm").set_attr()
OpsRegister.Register("dropout").set_attr()