MCPcopy Create free account
hub / github.com/pytorch/pytorch / conv

Function conv

caffe2/python/helpers/conv.py:167–186  ·  view source on GitHub ↗

2-dimensional convolution.

(
    model,
    blob_in,
    blob_out,
    dim_in,
    dim_out,
    kernel,
    weight_init=None,
    bias_init=None,
    WeightInitializer=None,
    BiasInitializer=None,
    group=1,
    transform_inputs=None,
    **kwargs
)

Source from the content-addressed store, hash-verified

165
166
167def conv(
168 model,
169 blob_in,
170 blob_out,
171 dim_in,
172 dim_out,
173 kernel,
174 weight_init=None,
175 bias_init=None,
176 WeightInitializer=None,
177 BiasInitializer=None,
178 group=1,
179 transform_inputs=None,
180 **kwargs
181):
182 """2-dimensional convolution.
183 """
184 return _ConvBase(model, False, blob_in, blob_out, dim_in, dim_out, kernel,
185 weight_init, bias_init, WeightInitializer, BiasInitializer,
186 group, transform_inputs, **kwargs)
187
188
189def conv_transpose(

Callers 4

group_convFunction · 0.70
lc_2d_nchwMethod · 0.50
lc_1d_nchwMethod · 0.50
lc_3d_nchwMethod · 0.50

Calls 1

_ConvBaseFunction · 0.85

Tested by 3

lc_2d_nchwMethod · 0.40
lc_1d_nchwMethod · 0.40
lc_3d_nchwMethod · 0.40

Used in the wild real call sites across dependent graphs

searching dependent graphs…