xref: /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/Types.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 
3 #include <ATen/miopen/miopen-wrapper.h>
4 #include <ATen/Tensor.h>
5 
6 namespace at { namespace native {
7 
8 miopenDataType_t getMiopenDataType(const at::Tensor& tensor);
9 
10 int64_t miopen_version();
11 
12 }}  // namespace at::miopen
13