29 #include <unordered_set>
55 extern const std::unordered_set<BinaryEWOpCode, utility::hash_enum_class>
68 #ifdef BUILD_CUDA_MODULE
69 void BinaryEWCUDA(
const Tensor& lhs,
void Div(const Tensor &lhs, const Tensor &rhs, Tensor &dst)
Definition: BinaryEW.h:87
void BinaryEW(const Tensor &lhs, const Tensor &rhs, Tensor &dst, BinaryEWOpCode op_code)
Definition: BinaryEW.cpp:48
void Mul(const Tensor &lhs, const Tensor &rhs, Tensor &dst)
Definition: BinaryEW.h:83
void Add(const Tensor &lhs, const Tensor &rhs, Tensor &dst)
Definition: BinaryEW.h:75
void Sub(const Tensor &lhs, const Tensor &rhs, Tensor &dst)
Definition: BinaryEW.h:79
BinaryEWOpCode
Definition: BinaryEW.h:39
const std::unordered_set< BinaryEWOpCode, utility::hash_enum_class > s_boolean_binary_ew_op_codes
Definition: BinaryEW.cpp:40
void BinaryEWCPU(const Tensor &lhs, const Tensor &rhs, Tensor &dst, BinaryEWOpCode op_code)
Definition: BinaryEWCPU.cpp:160
Definition: PinholeCameraIntrinsic.cpp:35