HomeSort by relevance Sort by last modified time
    Searched refs:tanh_op (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops_gpu.cu.cc 99 Eigen::internal::scalar_tanh_op<T> tanh_op; local
112 tanh_op(icfo[1 * cell_size + gid] + b[1 * cell_size + act_id]);
131 const T co_local = tanh_op(cs_local);
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 671 auto* tanh_op = tensorflow_graph->add_node(); local
672 tanh_op->set_op("Tanh");
673 tanh_op->set_name(src_op.outputs[0]);
674 *tanh_op->add_input() = src_op.inputs[0];
675 (*tanh_op->mutable_attr())["T"].set_type(DT_FLOAT);
    [all...]

Completed in 80 milliseconds