Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for FusedBatchNormGrad (0.2 sec)

  1. 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)
Back to top