1#pragma once
2
3#include <c10/core/MemoryFormat.h>
4#include <torch/csrc/Export.h>
5#include <torch/csrc/utils/python_stub.h>
6
7namespace torch {
8namespace utils {
9
10void initializeMemoryFormats();
11TORCH_PYTHON_API PyObject* getTHPMemoryFormat(c10::MemoryFormat);
12
13} // namespace utils
14} // namespace torch
15