forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathvararg_functions.h
44 lines (29 loc) · 1023 Bytes
/
vararg_functions.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#pragma once
#include <ATen/core/List.h>
#include <ATen/core/functional.h>
#include <ATen/core/ivalue.h>
#include <ATen/core/jit_type.h>
#include <ATen/core/stack.h>
namespace torch {
namespace jit {
void tupleUnpack(Stack& stack);
void format(Stack& stack, size_t num_inputs);
void einsum(Stack& stack, size_t num_inputs);
void percentFormat(Stack& stack, size_t num_inputs);
void listUnpack(Stack& stack, size_t num_outputs);
void tupleConstruct(Stack& stack, size_t num_inputs);
void namedTupleConstruct(
Stack& stack,
at::TupleTypePtr type,
size_t num_inputs);
void listConstruct(
Stack& stack,
const at::ListType& list_type,
size_t num_inputs);
void dictConstruct(Stack& stack, const at::DictType& type, size_t num_inputs);
void createObject(Stack& stack, const at::ClassTypePtr& type);
void isinstance(Stack& stack, at::ArrayRef<at::TypePtr> types);
void tupleSlice(Stack& stack, size_t begin, size_t end);
void dequantize(Stack& stack);
} // namespace jit
} // namespace torch