24.02.1
|
#include <GpuElementwiseBinaryCommon.h>
Public Types | |
enum | ElementwiseOp { Add, Sub, Div, Mul, Min, Max, SquaredDiff, Power, Prelu } |
Public Member Functions | |
ElementwiseBinaryCommonAttributes & | operation (const ElementwiseBinaryCommonAttributes::ElementwiseOp &operation) |
Set operation. More... | |
ElementwiseOp | operation () const |
Get operation. More... | |
Definition at line 38 of file GpuElementwiseBinaryCommon.h.
|
strong |
Enumerator | |
---|---|
Add | (x + y) |
Sub | (x - y) |
Div | (x / y) |
Mul | (x * y) |
Min | Min(x, y) |
Max | Max(x, y) |
SquaredDiff | (x - y)^2 |
Power | x ^ y |
Prelu | y*x if x < 0, x otherwise |
Definition at line 41 of file GpuElementwiseBinaryCommon.h.
ElementwiseBinaryCommonAttributes::ElementwiseOp operation | ( | ) | const |
Get operation.
Definition at line 102 of file GpuElementwiseBinaryCommon.cpp.
Referenced by ElementwiseBinaryCommonAttributes::operation().
ElementwiseBinaryCommonAttributes & operation | ( | const ElementwiseBinaryCommonAttributes::ElementwiseOp & | operation | ) |
Set operation.
Definition at line 96 of file GpuElementwiseBinaryCommon.cpp.
References ElementwiseBinaryCommonAttributes::operation().
Referenced by GpuAdd::create_op(), GpuMul::create_op(), GpuSub::create_op(), GpuCkwElementwiseBinary::get_name(), GpuAdd::is_supported_op(), GpuMul::is_supported_op(), GpuSub::is_supported_op(), arm_compute::experimental::dynamic_fusion::to_ckw(), ClComponentElementwiseBinary::validate(), GpuAdd::validate_op(), GpuSub::validate_op(), and GpuMul::validate_op().