Searched refs:post_op (Results 1 – 8 of 8) sorted by relevance
138 const auto &post_op = post_ops.get_list().at(post_op_id); in set_post_ops_cl_build_options() local140 if(post_op->type() == experimental::PostOpType::Activation) in set_post_ops_cl_build_options()142 …= utils::cast::polymorphic_downcast<const experimental::PostOpAct<ITensorInfo *> *>(post_op.get()); in set_post_ops_cl_build_options()150 else if(post_op->type() == experimental::PostOpType::Eltwise_Add) in set_post_ops_cl_build_options()153 …= slot_prefix + "_ELTWISE_OP=ADD" + "_X_POS_" + support::cpp11::to_string(post_op->prev_dst_pos()); in set_post_ops_cl_build_options()155 for(const auto &tensor : post_op->arguments()) in set_post_ops_cl_build_options()164 else if(post_op->type() == experimental::PostOpType::Eltwise_PRelu) in set_post_ops_cl_build_options()167 …slot_prefix + "_ELTWISE_OP=PRELU" + "_X_POS_" + support::cpp11::to_string(post_op->prev_dst_pos()); in set_post_ops_cl_build_options()169 for(const auto &tensor : post_op->arguments()) in set_post_ops_cl_build_options()
46 for(const auto &post_op : post_ops.get_list()) in transform_post_op_list_arguments() local48 switch(post_op->type()) in transform_post_op_list_arguments()52 …t auto _post_op = utils::cast::polymorphic_downcast<const PostOpAct<FromTensorT> *>(post_op.get()); in transform_post_op_list_arguments()58 …_post_op = utils::cast::polymorphic_downcast<const PostOpEltwiseAdd<FromTensorT> *>(post_op.get()); in transform_post_op_list_arguments()64 …ost_op = utils::cast::polymorphic_downcast<const PostOpEltwisePRelu<FromTensorT> *>(post_op.get()); in transform_post_op_list_arguments()
47 for(auto &post_op : post_ops.get_list()) in post_ops() local49 switch(post_op->type()) in post_ops()53 …utils::cast::polymorphic_downcast<const experimental::PostOpAct<SimpleTensor<T>> *>(post_op.get()); in post_ops()59 …cast::polymorphic_downcast<const experimental::PostOpEltwiseAdd<SimpleTensor<T>> *>(post_op.get()); in post_ops()65 …st::polymorphic_downcast<const experimental::PostOpEltwisePRelu<SimpleTensor<T>> *>(post_op.get()); in post_ops()
288 string post_op = fused_ops_[1]; in ExtendMklDnnMatMulFwdParams() local290 if (post_op == "Relu") { in ExtendMklDnnMatMulFwdParams()292 } else if (post_op == "Relu6") { in ExtendMklDnnMatMulFwdParams()294 } else if (post_op == "Elu") { in ExtendMklDnnMatMulFwdParams()296 } else if (post_op == "GeluApproximate") { in ExtendMklDnnMatMulFwdParams()298 } else if (post_op == "GeluExact") { in ExtendMklDnnMatMulFwdParams()300 } else if (post_op == "Tanh") { in ExtendMklDnnMatMulFwdParams()302 } else if (post_op == "Add") { in ExtendMklDnnMatMulFwdParams()304 } else if (post_op == "LeakyRelu") { in ExtendMklDnnMatMulFwdParams()307 } else if (post_op == "Sigmoid") { in ExtendMklDnnMatMulFwdParams()[all …]
22 const struct xnn_post_operation post_op = post_operations[i]; in allocate_and_initialize_post_operation_params() local23 switch (post_op.op_type) { in allocate_and_initialize_post_operation_params()37 const struct xnn_post_operation post_op = post_operations[i]; in allocate_and_initialize_post_operation_params() local38 switch (post_op.op_type) { in allocate_and_initialize_post_operation_params()
383 for(const auto &post_op : post_op_node_list) in fuse_convolution_with_post_op() local385 switch(post_op->type()) in fuse_convolution_with_post_op()389 … auto *eltwise_node = arm_compute::utils::cast::polymorphic_downcast<EltwiseLayerNode *>(post_op); in fuse_convolution_with_post_op()393 … ARM_COMPUTE_LOG_GRAPH_VERBOSE(" with Elementwise Layer node with ID : " << post_op->id()); in fuse_convolution_with_post_op()398 … auto *act_node = arm_compute::utils::cast::polymorphic_downcast<ActivationLayerNode *>(post_op); in fuse_convolution_with_post_op()402 … ARM_COMPUTE_LOG_GRAPH_VERBOSE(" with Activation Layer node with ID : " << post_op->id()); in fuse_convolution_with_post_op()413 transfer_driving_nodes_and_remove_old_node(g, fused_node, post_op, true); in fuse_convolution_with_post_op()418 g.remove_node(post_op->id()); in fuse_convolution_with_post_op()750 for(auto &post_op : post_op_node_list) in fuse_convolution_batch_normalization_with_post_ops() local752 post_ops_name += post_op->name(); in fuse_convolution_batch_normalization_with_post_ops()
206 inline ::std::ostream &operator<<(::std::ostream &os, const experimental::IPostOp<T> &post_op)209 os << post_op.type() << ",";210 os << "prev_dst_pos=" << post_op.prev_dst_pos() << ",";211 switch(post_op.type())215 …st auto _post_op = utils::cast::polymorphic_downcast<const experimental::PostOpAct<T> *>(&post_op);221 … _post_op = utils::cast::polymorphic_downcast<const experimental::PostOpEltwiseAdd<T> *>(&post_op);227 …post_op = utils::cast::polymorphic_downcast<const experimental::PostOpEltwisePRelu<T> *>(&post_op);247 inline std::string to_string(const experimental::IPostOp<T> &post_op) in to_string() argument250 str << post_op; in to_string()264 for(const auto &post_op : post_ops.get_list()) variable[all …]
454 …ilter_node* child;} root; struct {__anonymous_enum__ type; ast_link_type post_op; ast_link_type pr…457 …type of 'struct {__anonymous_enum__ type; ast_link_type post_op; ast_link_type pre_op; union {char…462 type of 'ast_link_type post_op' changed: