#! /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"
#! /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()
#! /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())