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')

#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(