Class operator_desc
-
namespace jd
Note
jit classes in this file may not meet the coding convention and is to be refactored.
-
class operator_desc
- #include <operator_desc.hpp>
The operator descriptor class, describing a specific kind of operator.
Public Functions
-
inline operator_desc()
-
inline operator_desc(const kernel_kind &ker_kind, const kernel_prop &ker_prop, const engine_kind &eng_kind, const std::vector<tensor_desc> &ts_descs, const std::unordered_map<std::string, std::string> &attrs, const std::vector<postop_attr> &apply_postops_list = {})
-
inline operator_desc(const kernel_kind &ker_kind, const kernel_prop &ker_prop, const engine_kind &eng_kind, const runtime_kind &runtime_kind, const std::vector<tensor_desc> &ts_descs, const std::unordered_map<std::string, std::string> &attrs, const std::vector<postop_attr> &apply_postops_list = {})
-
inline virtual ~operator_desc()
-
inline bool operator==(const operator_desc &rhs) const
-
inline void set_binaryop_list(const std::vector<binaryop_attr> &binaryop_list)
-
inline const uint64_t &impl_nthr() const
-
inline const std::vector<tensor_desc> &tensor_descs() const
-
inline const std::unordered_map<std::string, std::string> &attrs() const
-
inline const std::vector<postop_attr> &apply_postops_list() const
-
inline const std::vector<binaryop_attr> &get_binaryop_list() const
-
inline std::vector<std::vector<dim_t>> tensor_shapes() const
-
inline std::vector<data_type> tensor_dtypes() const
-
inline std::vector<format_type> tensor_ftypes() const
Private Members
-
jd::kernel_prop ker_prop_ = kernel_prop::forward_inference
-
jd::engine_kind engine_kind_ = engine_kind::cpu
-
jd::runtime_kind runtime_kind_ = runtime_kind::undef
-
uint64_t impl_nthr_
-
std::vector<tensor_desc> ts_descs_
-
std::unordered_map<std::string, std::string> attrs_
-
std::vector<postop_attr> apply_postops_list_
-
std::vector<binaryop_attr> binaryop_list_
-
inline operator_desc()
-
class operator_desc