operator.h 293 Bytes
Newer Older
PanZezhongQY's avatar
PanZezhongQY committed
1
2
3
4
5
6
7
8
9
10
11
12
13
#ifndef __INFINIOP_OPERATOR___
#define __INFINIOP_OPERATOR___

#include "./handle.h"
#include "./tensor_descriptor.h"

// Base descriptor for all operators
typedef struct InfiniopDescriptor {
    infiniDevice_t device;
    int device_id;
} InfiniopDescriptor;

#endif //__INFINIOP_OPERATOR___