Home
last modified time | relevance | path

Searched refs:PreprocessBatchNorm (Results 1 – 5 of 5) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/
H A Dnnvm_to_onnx-inl.h230 void PreprocessBatchNorm(const NodeAttrs &attrs,
235 {"BatchNorm", PreprocessBatchNorm}
H A Dnnvm_to_onnx.cc919 void PreprocessBatchNorm(const NodeAttrs &attrs, in PreprocessBatchNorm() function
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/
H A Dnnvm_to_onnx-inl.h230 void PreprocessBatchNorm(const NodeAttrs &attrs,
235 {"BatchNorm", PreprocessBatchNorm}
H A Dnnvm_to_onnx.cc919 void PreprocessBatchNorm(const NodeAttrs &attrs, in PreprocessBatchNorm() function
/dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/routines/
H A Dnormalization.cc61 PreprocessBatchNormResult PreprocessBatchNorm( in PreprocessBatchNorm() function
276 …PreprocessBatchNormResult result = PreprocessBatchNorm(x, gamma, beta, running_mean, running_var, … in BatchNorm()
417PreprocessBatchNorm(x, gamma.AsGradStopped(), beta.AsGradStopped(), mean.AsGradStopped(), var.AsGr… in FixedBatchNorm()