OP_REQUIRES_OK(context, TF_TensorToTensor(tmp_var_tf.get(), &tmp_var->val));
  OP_REQUIRES_OK(context,
                 context->step_container()->Create(rm, unique_name, tmp_var));
  context->set_output_ref(0, &tmp_var->mu, &tmp_var->val);

  if (context->track_allocations()) {
    context->record_persistent_memory_allocation(tmp_var->val.AllocatedBytes());
  }

  TF_SetStatus(tf_status, TF_OK, "");
}