"googlemock/test/gmock-generated-matchers_test.cc" did not exist on "db22c227826b82e1ad05d6c47facfef73c99e057"
relu.h 822 Bytes
Newer Older
PanZezhongQY's avatar
PanZezhongQY committed
1
2
3
4
5
6
7
#ifndef __INFINIOP_RELU_H__
#define __INFINIOP_RELU_H__

#include "../operator.h"

typedef InfiniopDescriptor *infiniopReluDescriptor_t;

PanZezhong's avatar
PanZezhong committed
8
9
10
11
__C __export infiniStatus_t infiniopCreateReluDescriptor(infiniopHandle_t handle,
                                                         infiniopReluDescriptor_t *desc_ptr,
                                                         infiniopTensorDescriptor_t y,
                                                         infiniopTensorDescriptor_t x);
PanZezhongQY's avatar
PanZezhongQY committed
12

PanZezhong's avatar
PanZezhong committed
13
14
15
16
__C __export infiniStatus_t infiniopRelu(infiniopReluDescriptor_t desc,
                                         void *y,
                                         void const *x,
                                         void *stream);
PanZezhongQY's avatar
PanZezhongQY committed
17

PanZezhong's avatar
PanZezhong committed
18
__C __export infiniStatus_t infiniopDestroyReluDescriptor(infiniopReluDescriptor_t desc);
PanZezhongQY's avatar
PanZezhongQY committed
19
20

#endif