Searched refs:softmax_op (Results 1 – 5 of 5) sorted by relevance
49 "//tensorflow/core/kernels:softmax_op",
87 "softmax_op.cc",
839 tensorflow::NodeDef* softmax_op = tensorflow_graph->add_node(); in ConvertSoftmaxOperator() local840 softmax_op->set_op("Softmax"); in ConvertSoftmaxOperator()841 softmax_op->set_name(src_op.outputs[0]); in ConvertSoftmaxOperator()842 *softmax_op->add_input() = softmax_input; in ConvertSoftmaxOperator()845 (*softmax_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertSoftmaxOperator()
243 tensorflow/core/kernels/softmax_op.cc
3871 ":softmax_op",3949 name = "softmax_op",3950 prefix = "softmax_op",5545 "softmax_op.cc",