Home
last modified time | relevance | path

Searched refs:tanh_op (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops_gpu.cu.cc145 Eigen::internal::scalar_tanh_op<T> tanh_op; in lstm_gates() local
158 tanh_op(icfo[1 * cell_size + gid] + b[1 * cell_size + act_id]); in lstm_gates()
177 const T co_local = tanh_op(cs_local); in lstm_gates()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc802 tensorflow::NodeDef* tanh_op = tensorflow_graph->add_node(); in ConvertTanhOperator() local
803 tanh_op->set_op("Tanh"); in ConvertTanhOperator()
804 tanh_op->set_name(src_op.outputs[0]); in ConvertTanhOperator()
805 *tanh_op->add_input() = src_op.inputs[0]; in ConvertTanhOperator()
806 (*tanh_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertTanhOperator()