| 79 | {"scale", core::PortKind::Parameter, false}, |
| 80 | {"bias", core::PortKind::Parameter, true}, |
| 81 | }; |
| 82 | |
| 83 | const core::ModuleSchema kBatchNorm1dEvalSchema = { |
| 84 | "BatchNorm1dEval", |
| 85 | "nn.normalization", |
| 86 | kBatchNorm1dEvalInputs, |
| 87 | 3, |
| 88 | kNormOutputs, |
| 89 | 1, |
| 90 | "Applies precomputed 1D batch-normalization eval scale and bias to channel-first tensors.", |
| 91 | }; |
| 92 | |
| 93 | core::TensorValue ensure_f32( |
| 94 | core::ModuleBuildContext & ctx, |
no outgoing calls
no test coverage detected