| H A D | core_named_ops.py | 240 def conv_3d(I=TensorDef(T1, S.OD + S.KD, S.OH + S.KH, S.OW + S.KW), argument 242 O=TensorDef(U, S.OD, S.OH, S.OW, output=True)): 392 def conv_3d_ndhwc_dhwcf(I=TensorDef(T1, S.N, S.OD * S.SD + S.KD * S.DD, argument 611 O=TensorDef(U, S.N, S.OD, S.OH, S.OW, 637 S.N, S.OD * S.SD + S.KD * S.DD, argument 641 O=TensorDef(U, S.N, S.OD, S.OH, S.OW, S.CM, 809 def pooling_ndhwc_sum(I=TensorDef(T1, S.N, S.OD * S.SD + S.KD * S.DD, argument 817 O=TensorDef(U, S.N, S.OD, S.OH, S.OW, S.C, output=True), 836 def pooling_ndhwc_max(I=TensorDef(T1, S.N, S.OD * S.SD + S.KD * S.DD, argument 844 O=TensorDef(U, S.N, S.OD, S.OH, S.OW, S.C, output=True), [all …]
|