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