if (shape_arr[i] != 1 && stride_arr[i] != expected_stride) {
      valid = false;
    }
    expected_stride *= shape_arr[i];
  }
  return valid;
}
}  // namespace

void* TFE_GetDLDevice(TFE_TensorHandle* h, TF_Status* status) {
  auto dl_device = GetDlContext(h, status);
  return new DLDevice{dl_device.device_type, dl_device.device_id};
}

void TFE_CallDLManagedTensorDeleter(void* dlm_ptr) {