Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc880 tensorflow::NodeDef* log_softmax_op = tensorflow_graph->add_node(); in ConvertLogSoftmaxOperator() local
881 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()