mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
0d6eb209e6
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/21648 ghimport-source-id: 583f155c8ee95967d2f8b9d8df27d94b9e725694 Differential Revision: D15804482 Pulled By: zou3519 fbshipit-source-id: f86520dda479100be2a752e4db8a902167413a83
9 lines
159 B
C
9 lines
159 B
C
#pragma once
|
|
#ifdef NAMEDTENSOR_ENABLED
|
|
#include <torch/csrc/python_headers.h>
|
|
#include <ATen/Dimname.h>
|
|
|
|
at::Dimname THPDimname_parse(PyObject* obj);
|
|
|
|
#endif
|