/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/passes/shape_analysis.h (463B)
#pragma once
#include
#include
#include
namespace torch {
namespace jit {
struct Graph;
TORCH_API void EraseShapeInformation(const std::shared_ptr& graph);
TORCH_API void PropagateInputShapes(const std::shared_ptr& graph);
TORCH_API bool mergeTypes(
ArrayRef lhs,
ArrayRef rhs,
ArrayRef outputs);
} // namespace jit
} // namespace torch