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 BinaryEW(const Tensor &lhs, const Tensor &rhs, Tensor &dst, BinaryEWOpCode op_code)
Definition: BinaryEW.cpp:48
const std::unordered_set< BinaryEWOpCode, utility::hash_enum_class > s_boolean_binary_ew_op_codes
Definition: BinaryEW.cpp:40
BinaryEWOpCode
Definition: BinaryEW.h:39
Definition: PinholeCameraIntrinsic.cpp:35
void BinaryEWCPU(const Tensor &lhs, const Tensor &rhs, Tensor &dst, BinaryEWOpCode op_code)
Definition: BinaryEWCPU.cpp:175