4 #include <torch/csrc/jit/fuser/partition_desc.h> 5 #include <torch/csrc/jit/fuser/tensor_desc.h> 6 #include <torch/csrc/utils/disallow_copy.h> 22 std::vector<TensorDesc> input_desc,
23 std::vector<TensorDesc> output_desc,
24 std::vector<PartitionDesc> chunk_desc,
25 std::vector<PartitionDesc> concat_desc,
27 : name_(std::move(name)),
28 code_(std::move(code)),
29 input_desc_(std::move(input_desc)),
30 output_desc_(std::move(output_desc)),
31 chunk_desc_(std::move(chunk_desc)),
32 concat_desc_(std::move(concat_desc)),
33 has_random_(has_random) {}
47 virtual void launch_raw(
const uint32_t numel, std::vector<void*>& arguments)
52 const std::string& name()
const {
55 const std::string& code()
const {
58 const std::vector<TensorDesc>& inputDesc()
const {
61 const std::vector<TensorDesc>& outputDesc()
const {
64 const std::vector<PartitionDesc>& chunkDesc()
const {
67 const std::vector<PartitionDesc>& concatDesc()
const {
70 bool hasRandom()
const {
75 const std::string name_;
76 const std::string code_;
77 const std::vector<TensorDesc> input_desc_;
78 const std::vector<TensorDesc> output_desc_;
83 const std::vector<PartitionDesc> chunk_desc_;
88 const std::vector<PartitionDesc> concat_desc_;
90 const bool has_random_;
Backend
This legacy enum class defines the set of backends supported by old school, code generated Type-based...