| /external/tensorflow/tensorflow/core/kernels/ |
| cwise_op_atan2.cc | 19 REGISTER2(BinaryOp, CPU, "Atan2", functor::atan2, float, double);
21 REGISTER2(BinaryOp, GPU, "Atan2", functor::atan2, float, double);
|
| cwise_op_igammas.cc | 19 REGISTER2(BinaryOp, CPU, "Igamma", functor::igamma, float, double);
20 REGISTER2(BinaryOp, CPU, "Igammac", functor::igammac, float, double);
|
| cwise_op_zeta.cc | 19 REGISTER2(BinaryOp, CPU, "Zeta", functor::zeta, float, double);
20 REGISTER2(BinaryOp, CPU, "Polygamma", functor::polygamma, float, double);
|
| cwise_op_pow.cc | 19 REGISTER5(BinaryOp, CPU, "Pow", functor::pow, float, Eigen::half, double,
21 REGISTER2(BinaryOp, CPU, "Pow", functor::safe_pow, int32, int64);
24 REGISTER4(BinaryOp, GPU, "Pow", functor::pow, float, Eigen::half, double,
28 REGISTER2(BinaryOp, SYCL, "Pow", functor::pow, float, double);
|
| cwise_op_mod.cc | 19 REGISTER2(BinaryOp, CPU, "Mod", functor::safe_mod, int32, int64);
20 REGISTER2(BinaryOp, CPU, "Mod", functor::fmod, float, double);
21 REGISTER2(BinaryOp, CPU, "TruncateMod", functor::safe_mod, int32, int64);
22 REGISTER2(BinaryOp, CPU, "TruncateMod", functor::fmod, float, double);
34 BinaryOp<CPUDevice, functor::safe_mod<int32>>);
41 BinaryOp<CPUDevice, functor::safe_mod<int32>>);
|
| cwise_op_logical_and.cc | 20 BinaryOp<CPUDevice, functor::logical_and>);
23 BinaryOp<GPUDevice, functor::logical_and>);
|
| cwise_op_logical_or.cc | 20 BinaryOp<CPUDevice, functor::logical_or>);
23 BinaryOp<GPUDevice, functor::logical_or>);
|
| cwise_op_not_equal_to_1.cc | 19 REGISTER6(BinaryOp, CPU, "NotEqual", functor::not_equal_to, float, Eigen::half,
22 REGISTER4(BinaryOp, GPU, "NotEqual", functor::not_equal_to, float, Eigen::half,
|
| cwise_op_maximum.cc | 19 REGISTER6(BinaryOp, CPU, "Maximum", functor::maximum, float, Eigen::half,
22 REGISTER4(BinaryOp, GPU, "Maximum", functor::maximum, float, Eigen::half,
34 BinaryOp<CPUDevice, functor::maximum<int32>>);
38 REGISTER3(BinaryOp, SYCL, "Maximum", functor::maximum, float, double, int64);
45 BinaryOp<CPUDevice, functor::maximum<int32>>);
|
| cwise_op_minimum.cc | 19 REGISTER5(BinaryOp, CPU, "Minimum", functor::minimum, float, Eigen::half,
22 REGISTER4(BinaryOp, GPU, "Minimum", functor::minimum, float, Eigen::half,
34 BinaryOp<CPUDevice, functor::minimum<int32>>);
38 REGISTER3(BinaryOp, SYCL, "Minimum", functor::minimum, float, double, int64);
45 BinaryOp<CPUDevice, functor::minimum<int32>>);
|
| cwise_op_squared_difference.cc | 19 REGISTER5(BinaryOp, CPU, "SquaredDifference", functor::squared_difference,
22 REGISTER4(BinaryOp, GPU, "SquaredDifference", functor::squared_difference,
36 BinaryOp<CPUDevice, functor::squared_difference<int32>>);
39 REGISTER3(BinaryOp, SYCL, "SquaredDifference", functor::squared_difference,
48 BinaryOp<CPUDevice, functor::squared_difference<int32>>);
|
| cwise_op_div.cc | 19 REGISTER5(BinaryOp, CPU, "Div", functor::div, float, Eigen::half, double,
21 REGISTER5(BinaryOp, CPU, "Div", functor::safe_div, uint8, uint16, int16, int32,
23 REGISTER5(BinaryOp, CPU, "TruncateDiv", functor::safe_div, uint8, uint16, int16,
25 REGISTER5(BinaryOp, CPU, "RealDiv", functor::div, float, Eigen::half, double,
28 REGISTER9(BinaryOp, GPU, "Div", functor::div, float, Eigen::half, double, uint8,
30 REGISTER4(BinaryOp, GPU, "TruncateDiv", functor::div, uint8, uint16, int16,
32 REGISTER5(BinaryOp, GPU, "RealDiv", functor::div, float, Eigen::half, double,
44 BinaryOp<CPUDevice, functor::safe_div<int32>>);
48 REGISTER2(BinaryOp, SYCL, "Div", functor::div, float, double);
49 REGISTER2(BinaryOp, SYCL, "RealDiv", functor::div, float, double)
[all...] |
| cwise_op_sub.cc | 19 REGISTER7(BinaryOp, CPU, "Sub", functor::sub, float, Eigen::half, double, int32,
23 REGISTER4(BinaryOp, CPU, "Sub", functor::sub, int8, uint8, int16, uint16);
28 REGISTER(BinaryOp, CPU, "Sub", functor::sub, int32);
32 REGISTER6(BinaryOp, GPU, "Sub", functor::sub, float, Eigen::half, double, int64,
44 BinaryOp<CPUDevice, functor::sub<int32>>);
48 REGISTER3(BinaryOp, SYCL, "Sub", functor::sub, float, double, int64);
55 BinaryOp<CPUDevice, functor::sub<int32>>);
|
| cwise_op_add_1.cc | 19 REGISTER5(BinaryOp, CPU, "Add", functor::add, float, Eigen::half, double, int32,
21 REGISTER5(BinaryOp, CPU, "AddV2", functor::add, float, Eigen::half, double,
25 REGISTER3(BinaryOp, GPU, "Add", functor::add, float, Eigen::half, double);
26 REGISTER3(BinaryOp, GPU, "AddV2", functor::add, float, Eigen::half, double);
37 BinaryOp<CPUDevice, functor::add<int32>>);
44 BinaryOp<CPUDevice, functor::add<int32>>);
49 REGISTER(BinaryOp, SYCL, "Add", functor::add, type); \
50 REEGISTER(BinaryOp, SYCL, "AddV2", functor::add, type);
60 BinaryOp<CPUDevice, functor::add<int32>>);
67 BinaryOp<CPUDevice, functor::add<int32>>)
[all...] |
| cwise_op_floor_div.cc | 19 REGISTER5(BinaryOp, CPU, "FloorDiv", functor::safe_floor_div, uint8, uint16,
21 REGISTER3(BinaryOp, CPU, "FloorDiv", functor::floor_div_real, float,
25 REGISTER4(BinaryOp, GPU, "FloorDiv", functor::floor_div, uint8, uint16, int16,
27 REGISTER3(BinaryOp, GPU, "FloorDiv", functor::floor_div_real, float,
41 BinaryOp<CPUDevice, functor::safe_floor_div<int32>>);
51 BinaryOp<CPUDevice, functor::safe_floor_div<int32>>);
|
| cwise_op_mul_1.cc | 20 REGISTER5(BinaryOp, CPU, "Mul", functor::mul, float, Eigen::half, double, uint8,
26 REGISTER(BinaryOp, CPU, "Mul", functor::mul, int32);
30 REGISTER4(BinaryOp, GPU, "Mul", functor::mul, float, Eigen::half, double,
41 BinaryOp<CPUDevice, functor::mul<int32>>);
45 REGISTER3(BinaryOp, SYCL, "Mul", functor::mul, float, double, uint8);
52 BinaryOp<CPUDevice, functor::mul<int32>>);
|
| cwise_op_add_2.cc | 25 REGISTER6(BinaryOp, CPU, "Add", functor::add, int8, int16, complex64, uint8,
29 REGISTER5(BinaryOp, CPU, "AddV2", functor::add, int8, int16, complex64, uint8,
32 REGISTER4(BinaryOp, GPU, "Add", functor::add, uint8, int64, complex64,
34 REGISTER4(BinaryOp, GPU, "AddV2", functor::add, uint8, int64, complex64,
|
| cwise_op_floor_mod.cc | 19 REGISTER2(BinaryOp, CPU, "FloorMod", functor::safe_floor_mod, int32, int64);
20 REGISTER2(BinaryOp, CPU, "FloorMod", functor::floor_fmod, float, double);
32 BinaryOp<CPUDevice, functor::safe_floor_mod<int32>>);
42 BinaryOp<CPUDevice, functor::safe_floor_mod<int32>>);
|
| cwise_op_equal_to_2.cc | 25 REGISTER6(BinaryOp, CPU, "Equal", functor::equal_to, int32, int64, complex64,
28 REGISTER6(BinaryOp, GPU, "Equal", functor::equal_to, int8, int16, int64,
|
| cwise_op_mul_2.cc | 25 REGISTER6(BinaryOp, CPU, "Mul", functor::mul, int8, uint16, int16, int64,
28 REGISTER6(BinaryOp, GPU, "Mul", functor::mul, int8, uint16, int16, int64,
|
| cwise_op_not_equal_to_2.cc | 25 REGISTER6(BinaryOp, CPU, "NotEqual", functor::not_equal_to, int32, int64,
28 REGISTER6(BinaryOp, GPU, "NotEqual", functor::not_equal_to, int8, int16, int64,
|
| cwise_op_greater.cc | 19 REGISTER8(BinaryOp, CPU, "Greater", functor::greater, float, Eigen::half,
22 REGISTER7(BinaryOp, GPU, "Greater", functor::greater, float, Eigen::half,
34 BinaryOp<CPUDevice, functor::greater<int32>>);
37 REGISTER2(BinaryOp, SYCL, "Greater", functor::greater, float, double);
45 BinaryOp<CPUDevice, functor::greater<int32>>);
|
| cwise_op_greater_equal.cc | 19 REGISTER8(BinaryOp, CPU, "GreaterEqual", functor::greater_equal, float,
22 REGISTER7(BinaryOp, GPU, "GreaterEqual", functor::greater_equal, float,
34 BinaryOp<CPUDevice, functor::greater_equal<int32>>);
38 REGISTER2(BinaryOp, SYCL, "GreaterEqual", functor::greater_equal, float,
47 BinaryOp<CPUDevice, functor::greater_equal<int32>>);
|
| cwise_op_less.cc | 19 REGISTER8(BinaryOp, CPU, "Less", functor::less, float, Eigen::half, double,
22 REGISTER7(BinaryOp, GPU, "Less", functor::less, float, Eigen::half, double,
34 BinaryOp<CPUDevice, functor::less<int32>>);
37 REGISTER3(BinaryOp, SYCL, "Less", functor::less, float, double, int64);
44 BinaryOp<CPUDevice, functor::less<int32>>);
|
| /frameworks/compile/mclinker/include/mcld/Script/ |
| BinaryOp.h | 1 //===- BinaryOp.h ---------------------------------------------------------===//
23 /** \class BinaryOP
28 class BinaryOp : public Operator {
32 BinaryOp() : Operator(Operator::BINARY, TYPE), m_Size(0) {
37 ~BinaryOp() {}
53 IntOperand* BinaryOp<Operator::MUL>::eval(const Module&,
56 IntOperand* BinaryOp<Operator::DIV>::eval(const Module&,
59 IntOperand* BinaryOp<Operator::MOD>::eval(const Module&,
62 IntOperand* BinaryOp<Operator::ADD>::eval(const Module&,
65 IntOperand* BinaryOp<Operator::SUB>::eval(const Module&
[all...] |
