- context (gpu_context_type)
 - axis (integer scalar)
 - device_is_not_cuda (bool scalar)