def test_test_ops(self):
    attr = tf.function(test_ops.test_attr)(tf.float32)
    self.assertAllClose(attr.numpy(), 100.0)


if __name__ == '__main__':
  os.environ['TF_MLIR_TFR_LIB_DIR'] = (
      'tensorflow/compiler/mlir/tfr/examples/customization')

    self.assertAllEqual(
        x,
        gen_resource_variable_ops.ReadVariableOp(
            resource=handle, dtype=dtypes.float32))


if __name__ == '__main__':
  os.environ['TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/resources'
  ops.enable_eager_execution()

                               tf.function(gen_pad_ops.new_mirror_pad_grad),
                               ops_defs._composite_mirror_pad_grad, kwargs_)


if __name__ == '__main__':
  os.environ[
      'TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/examples/pad'

    t3 = constant_op.constant([[-10.0, -10.0], [-10.0, -10.0]])
    sq = biasd_dense_elu(t1, t2, t3)
    self.assertAllClose(sq.numpy().reshape(-1), [-0.950213, 0, 5, 12])


if __name__ == '__main__':
  os.environ['TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/resources'
  ops.enable_eager_execution()

                               tf.function(gen_mnist_ops.new_max_pool),
                               ops_defs._composite_max_pool, kwargs)


if __name__ == '__main__':
  os.environ[
      'TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/examples/mnist'

def main(strategy):
  """Trains an MNIST model using the given tf.distribute.Strategy."""
  # TODO(fengliuai): put this in some automatically generated code.
  os.environ[
      'TF_MLIR_TFR_LIB_DIR'] = 'tensorflow/compiler/mlir/tfr/examples/mnist'

  ds_train = tfds.load('mnist', split='train', shuffle_files=True)
  ds_train = ds_train.shuffle(1024).batch(batch_size).prefetch(64)

#include "tensorflow/core/platform/stringpiece.h"
#include "tensorflow/core/util/env_var.h"
#include "tsl/platform/statusor.h"

namespace tensorflow {
namespace tfr {

const char* const kTFRLibEnv = "TF_MLIR_TFR_LIB_DIR";

absl::StatusOr<std::unique_ptr<TFRDecomposeContext>> TFRDecomposeContext::Get(
    mlir::MLIRContext* mlir_ctx) {
  Env* env = Env::Default();
  std::string tfr_lib_dir;
  TF_RETURN_IF_ERROR(ReadStringFromEnvVar(