Exemplos de make_reduction_op em Python

Exemplo n.º 1

Arquivo: ops_bridge.py Projeto: rsumner31/ngraph

def ReduceProd(onnx_node,
               ng_inputs):  # type: (NodeWrapper, List[TensorOp]) -> Op
    return make_reduction_op(ng.prod, onnx_node, ng_inputs[0])

Exemplo n.º 2

Arquivo: ops_bridge.py Projeto: rsumner31/ngraph

def ArgMax(onnx_node, ng_inputs):  # type: (NodeWrapper, List[TensorOp]) -> Op
    return make_reduction_op(ng.argmax, onnx_node, ng_inputs[0])

Exemplo n.º 3

Arquivo: ops_bridge.py Projeto: rsumner31/ngraph

def ReduceLogSumExp(onnx_node,
                    ng_inputs):  # type: (NodeWrapper, List[TensorOp]) -> Op
    op = ng.exp(ng_inputs[0])
    op = make_reduction_op(ng.sum, onnx_node, op)
    op = ng.log(op)
    return op