from tensorflow.python.trackable import autotrackable
from tensorflow.python.util import tf_export

# Type aliases for quant_opts_pb2 messages.
_QuantizationOptions = tf_export.tf_export(
    'quantization.experimental.QuantizationOptions'
)(quant_opts_pb2.QuantizationOptions)

_QuantizationMethod = tf_export.tf_export(
    'quantization.experimental.QuantizationMethod'
)(quant_opts_pb2.QuantizationMethod)

// .go_export section.
func readpkglist(shlibpath string) (pkgs []*load.Package) {
	var stk load.ImportStack
	if cfg.BuildToolchainName == "gccgo" {
		f, err := elf.Open(shlibpath)
		if err != nil {
			base.Fatal(fmt.Errorf("failed to open shared library: %v", err))
		}
		defer f.Close()
		sect := f.Section(".go_export")
		if sect == nil {

// Expm1 op patterns.
//===----------------------------------------------------------------------===//

// Expm1(x) = Exp(x) - 1
def LowerExp1mOp : Pat<
  (TF_Expm1Op $x),
  (TF_SubOp
    (TF_ExpOp $x),
    (TF_ConstOp (GetScalarOfType<1> $x))
  )>;

//===----------------------------------------------------------------------===//
// MatrixBandPart op patterns.

foreach Mapping = [
                   [TF_AbsOp, MHLO_AbsOp],
                   [TF_CeilOp, MHLO_CeilOp],
                   [TF_ComplexAbsOp, MHLO_AbsOp],
                   [TF_CosOp, MHLO_CosineOp],
                   [TF_ExpOp, MHLO_ExpOp],
                   [TF_Expm1Op, MHLO_Expm1Op],
                   [TF_ErfOp, MHLO_ErfOp],
                   [TF_FloorOp, MHLO_FloorOp],
                   [TF_ImagOp, MHLO_ImagOp],

def LegalizeUnique : Pat<(TF_UniqueOp $arg0),(TFL_UniqueOp $arg0)>;

def LegalizeFloorMod : Pat<(TF_FloorModOp $arg0, $arg1),
                           (TFL_FloorModOp $arg0, $arg1)>;
def LegalizeExp : Pat<(TF_ExpOp $arg0), (TFL_ExpOp $arg0)>;

def LegalizeLRN : Pat<
  (TF_LRNOp $arg0, $radius, F32Attr:$bias, F32Attr:$alpha, F32Attr:$beta),
  (TFL_LocalResponseNormalizationOp $arg0, (convertIntAttrTo32Bit $radius),

    indicating its preferred use in language bindings for TensorFlow. What was
    previously `TF_Session` has been renamed to `TF_DeprecatedSession`.
*   Renamed `TF_Port` to `TF_Output` in the C API.
*   Removes RegisterShape from public API. Use C++ shape function registration
    instead. indexing now starts from 1 instead of 0, and `bus_id==0` is used

containing metadata about the TPUEmbedding partitioner output and
the HBM size (in bytes) required for operation.}]>:$common_config
  );
}

def TF_ExpOp : TF_Op<"Exp", [Pure, TF_SameOperandsAndResultTypeResolveRef]> {
  let summary = [{
Computes exponential of x element-wise.  \\(y = e^x\\).
  }];

  let description = [{