Searched refs:topk_op (Results 1 – 5 of 5) sorted by relevance
1908 tensorflow::NodeDef* topk_op = tensorflow_graph->add_node(); in ConvertTopKV2Operator() local1909 topk_op->set_op("TopKV2"); in ConvertTopKV2Operator()1910 topk_op->set_name(src_op.outputs[0]); in ConvertTopKV2Operator()1912 *topk_op->add_input() = src_op.inputs[0]; in ConvertTopKV2Operator()1913 *topk_op->add_input() = src_op.inputs[1]; in ConvertTopKV2Operator()1916 (*topk_op->mutable_attr())["T"].set_type(data_type); in ConvertTopKV2Operator()1917 (*topk_op->mutable_attr())["sorted"].set_b(true); in ConvertTopKV2Operator()
100 "topk_op.cc",
3874 ":topk_op",3970 name = "topk_op",3971 srcs = ["topk_op.cc"],3972 hdrs = ["topk_op.h"],3974 "topk_op.h",5640 "topk_op.h",5834 "topk_op.cc",
283 tensorflow/core/kernels/topk_op.cc
4406 "//tensorflow/core/kernels:topk_op",