Python _maybe_cast_reduce_op_inputの例

コード例 #1

ファイル: symbolic_opset13.py プロジェクト: huaxz1986/pytorch

 def symbolic(g, self, dim=None, keepdim=None):
     self = opset9._maybe_cast_reduce_op_input(g, self)
     if dim is None:
         # all-reduce path
         return symbolic_helper._handle_reduce_dim_none(g, self, onnx_op_name)
     else:
         keepdim = symbolic_helper._get_const(keepdim, "i", "keepdim")
         return g.op(onnx_op_name, self, dim, keepdims_i=keepdim)

コード例 #2

ファイル: symbolic_opset13.py プロジェクト: xsacha/pytorch

 def symbolic(g, self, dim=None, keepdim=None):
     self = _maybe_cast_reduce_op_input(g, self)
     if dim is None:
         # all-reduce path
         return sym_help._handle_reduce_dim_none(g, self, onnx_op_name)
     else:
         keepdim = sym_help._get_const(keepdim, 'i', 'keepdim')
         return g.op(onnx_op_name, self, dim, keepdims_i=keepdim)

コード例 #3

ファイル: symbolic_opset13.py プロジェクト: sbrodehl/pytorch

 def symbolic(g, self, dim=None, keepdim=None):
     self = _maybe_cast_reduce_op_input(g, self)
     if dim is None:
         # all-reduce path
         return g.op(onnx_op_name, self, keepdims_i=0)
     else:
         keepdim = sym_help._get_const(keepdim, "i", "keepdim")
         return g.op(onnx_op_name, self, dim, keepdims_i=keepdim)