Searched refs:log_softmax_op (Results 1 – 1 of 1) sorted by relevance
887 tensorflow::NodeDef* log_softmax_op = tensorflow_graph->add_node(); in ConvertLogSoftmaxOperator() local888 log_softmax_op->set_op("LogSoftmax"); in ConvertLogSoftmaxOperator()889 log_softmax_op->set_name(src_op.outputs[0]); in ConvertLogSoftmaxOperator()890 *log_softmax_op->add_input() = softmax_input; in ConvertLogSoftmaxOperator()891 (*log_softmax_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertLogSoftmaxOperator()