10#include <unordered_set>
38extern const std::unordered_set<BinaryEWOpCode, utility::hash_enum_class>
51#ifdef BUILD_CUDA_MODULE
52void BinaryEWCUDA(
const Tensor& lhs,
void BinaryEW(const Tensor &lhs, const Tensor &rhs, Tensor &dst, BinaryEWOpCode op_code)
Definition: BinaryEW.cpp:29
BinaryEWOpCode
Definition: BinaryEW.h:20
const std::unordered_set< BinaryEWOpCode, utility::hash_enum_class > s_boolean_binary_ew_op_codes
Definition: BinaryEW.cpp:21
void BinaryEWCPU(const Tensor &lhs, const Tensor &rhs, Tensor &dst, BinaryEWOpCode op_code)
Definition: BinaryEWCPU.cpp:153
Tensor Minimum(const Tensor &input, const Tensor &other)
Computes the element-wise minimum of input and other. The tensors must have same data type and device...
Definition: TensorFunction.cpp:136
Tensor Maximum(const Tensor &input, const Tensor &other)
Computes the element-wise maximum of input and other. The tensors must have same data type and device...
Definition: TensorFunction.cpp:124
Definition: PinholeCameraIntrinsic.cpp:16