Searched refs:tanh_op (Results 1 – 3 of 3) sorted by relevance
145 Eigen::internal::scalar_tanh_op<T> tanh_op; in lstm_gates() local162 const T ci_local = tanh_op(gates[c_offset + gid] + b[c_offset + act_id]); in lstm_gates()181 const T co_local = tanh_op(cs_local); in lstm_gates()
807 tensorflow::NodeDef* tanh_op = tensorflow_graph->add_node(); in ConvertTanhOperator() local808 tanh_op->set_op("Tanh"); in ConvertTanhOperator()809 tanh_op->set_name(src_op.outputs[0]); in ConvertTanhOperator()810 *tanh_op->add_input() = src_op.inputs[0]; in ConvertTanhOperator()811 (*tanh_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertTanhOperator()
2662 auto tanh_op = rewriter.create<TanhOp>(loc, scaled_input); in matchAndRewrite() local2663 auto mul_op = rewriter.create<MulOp>(loc, tanh_op, half); in matchAndRewrite()