FunctionDefLibrary library;

  FunctionDef sort_func = FunctionDefHelper::Create(
      "SortFunc",
      /*in_def=*/{"x:float"},
      /*out_def=*/{"res:float"},
      /*attr_def=*/{},
      /*node_def=*/{{{"t0"}, "XlaSort", {"x"}, {{"T", DT_FLOAT}}}},
      /*ret_def*/ {{"res", "t0:output"}});

  *library.add_function() = sort_func;

  Output in = ops::Placeholder(root, DT_FLOAT);