Struct BatchNormHelper
Defined in File batch_norm.cc
Struct Documentation
-
struct
cinn::frontend::decomposer
::
BatchNormHelper
Public Functions
-
BatchNormHelper
(CinnBuilder *cinn_builder, const std::vector<int> &arg_x_shape, const std::vector<int> &arg_param_shape, std::string data_layout, std::string bn_op_type)
-
~BatchNormHelper
()
-