mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
5b3b2b9cc7
Follows #131986 Pull Request resolved: https://github.com/pytorch/pytorch/pull/131996 Approved by: https://github.com/ezyang
24 lines
504 B
C++
24 lines
504 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/ir/ir.h>
|
|
|
|
#include <vector>
|
|
|
|
namespace torch::jit {
|
|
|
|
TORCH_API TypePtr getTensorType(const at::Tensor& t, bool complete);
|
|
|
|
TORCH_API TypePtr inferShapeAndTypeForInput(
|
|
TypePtr input_type,
|
|
Stack::const_iterator& s_iter,
|
|
const Stack::const_iterator& s_iter_end,
|
|
bool complete);
|
|
|
|
TORCH_API void setInputTensorTypes(
|
|
Graph& g,
|
|
const Stack& stack,
|
|
bool complete,
|
|
const std::vector<int>& param_count_list = {});
|
|
|
|
} // namespace torch::jit
|