| 523 | } |
| 524 | |
| 525 | ValueRefList batchnorm_rule( |
| 526 | const BatchNorm& op, Span<ValueRef>& inputs, const bool& auto_convert, |
| 527 | const FormatTransformation& t) { |
| 528 | auto&& inp_format = inputs[0].cast(t.value_type()).format(); |
| 529 | if (inp_format == FT::NHWC) { |
| 530 | auto new_param = op.param(); |
| 531 | new_param.param_dim = BatchNorm::ParamDim::DIM_111C; |
| 532 | auto new_op = BatchNorm::make(new_param); |
| 533 | return identity_rule_helper(*new_op, inputs, t); |
| 534 | } |
| 535 | return identity_rule_helper(op, inputs, t); |
| 536 | } |
| 537 | |
| 538 | ValueRefList adaptive_pooling_rule( |
| 539 | const AdaptivePooling& op, Span<ValueRef>& inputs, const bool& auto_convert, |
nothing calls this directly
no test coverage detected