Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for FusedBatchNormGrad (2.06 sec)

  1. tensorflow/compiler/jit/mark_for_compilation_pass.cc

              {"REDUCEWINDOWPW", {"BiasAddGrad", "LRN", "LRNGrad"}},
              {"BN",
               {"FusedBatchNorm", "FusedBatchNormV2", "FusedBatchNormV3",
                "_FusedBatchNormEx", "FusedBatchNormGrad", "FusedBatchNormGradV2",
                "FusedBatchNormGradV3"}},
              {"Conv", {"_FusedConv2D"}},
              {"SORT", {"TopKV2"}},  // XLA version much faster then TF version.
              {"MISC",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-NEXT: %[[x_backprop:.*]] = mhlo.convert %[[mul3]] : tensor<8x8x8x8xf32>
      // CHECK-NEXT: return %[[x_backprop]] : tensor<8x8x8x8xf32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        return success();
      }
    };
    
    using ConvertRFFTOp = ConvertFFTOp<TF::RFFTOp>;
    using ConvertIRFFTOp = ConvertFFTOp<TF::IRFFTOp>;
    
    // The base class to convert TensorFlow FusedBatchNormGrad*Op to HLO
    // BatchNormGradOp for training and a sequence of binary ops for inference.
    // TODO(b/145536565): move to legalize_tf_patterns.td if it applies.
    template <typename FusedBatchNormGradOpT>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
    
      let hasCanonicalizer = 1;
    
      let hasVerifier = 1;
    }
    
    def TF_FusedBatchNormGradOp : TF_Op<"FusedBatchNormGrad", [Pure]> {
      let summary = "Gradient for batch normalization.";
    
      let description = [{
    Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW".
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top