FusedBatchNorm that can be replaced with a cheaper set of primitives.
| 93 | |
| 94 | // FusedBatchNorm that can be replaced with a cheaper set of primitives. |
| 95 | struct FusedBatchNorm { |
| 96 | FusedBatchNorm() = default; |
| 97 | explicit FusedBatchNorm(int fused_batch_norm) |
| 98 | : fused_batch_norm(fused_batch_norm) {} |
| 99 | |
| 100 | int fused_batch_norm = kMissingIndex; |
| 101 | }; |
| 102 | |
| 103 | // FusedBatchNorm[$is_training] with fused side input and/or activation. |
| 104 | struct FusedBatchNormEx { |
no outgoing calls