Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for FusedBatchNormV3Pat (0.33 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

    //      (HasNoUseOf:$root__3), (HasNoUseOf:$root__4),
    //      (HasNoUseOf:$root__5), (AreBroadcastableTypes $multiplier, $x)]>;
    
    struct FusedBatchNormV3Pat : public ::mlir::RewritePattern {
      explicit FusedBatchNormV3Pat(::mlir::MLIRContext *context)
          : ::mlir::RewritePattern(
                "tf.FusedBatchNormV3", 1, context,
                {"tf.Add", "tf.Const", "tf.Mul", "tf.Rsqrt", "tf.Sub"}) {}
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
Back to top