Home
last modified time | relevance | path

Searched refs:is_batchnorm3d_module (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/quantization/
Dhelper.h211 bool is_batchnorm3d_module(
Dinsert_observers.cpp798 {is_batchnorm3d_module, is_relu_module});
806 {is_batchnorm3d_module, is_functional_relu});
814 {is_batchnorm3d_module});
822 {is_batchnorm3d_module});
Dhelper.cpp788 bool is_batchnorm3d_module( in is_batchnorm3d_module() function
/external/pytorch/torch/csrc/jit/passes/
Dfold_conv_bn.cpp397 {is_conv3d_module, is_batchnorm3d_module}); in FoldConvBatchNorm()