Module caffe2.python.helpers.normalization. More...
Functions | |
| def | LRN (model, blob_in, blob_out, kwargs) |
| def | Softmax (model, blob_in, blob_out=None, use_cudnn=False, kwargs) |
| def | InstanceNorm (model, blob_in, blob_out, dim_in, order="NCHW", kwargs) |
| def | SpatialBN (model, blob_in, blob_out, dim_in, order="NCHW", kwargs) |
Module caffe2.python.helpers.normalization.
| def normalization.LRN | ( | model, | |
| blob_in, | |||
| blob_out, | |||
| kwargs | |||
| ) |
LRN
Definition at line 11 of file normalization.py.
| def normalization.Softmax | ( | model, | |
| blob_in, | |||
blob_out = None, |
|||
use_cudnn = False, |
|||
| kwargs | |||
| ) |
Softmax.
Definition at line 21 of file normalization.py.
1.8.14