Searched refs:topk_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1942 tensorflow::NodeDef* topk_op = tensorflow_graph->add_node(); in ConvertTopKV2Operator() local
1943 topk_op->set_op("TopKV2"); in ConvertTopKV2Operator()
1944 topk_op->set_name(src_op.outputs[0]); in ConvertTopKV2Operator()
1946 *topk_op->add_input() = src_op.inputs[0]; in ConvertTopKV2Operator()
1947 *topk_op->add_input() = src_op.inputs[1]; in ConvertTopKV2Operator()
1950 (*topk_op->mutable_attr())["T"].set_type(data_type); in ConvertTopKV2Operator()
1951 (*topk_op->mutable_attr())["sorted"].set_b(true); in ConvertTopKV2Operator()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 118 "topk_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3896 ":topk_op",
4000 name = "topk_op",
4001 srcs = ["topk_op.cc"],
4002 hdrs = ["topk_op.h"],
4004 "topk_op.h",
5962 "topk_op.h",
6270 "topk_op.cc",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5688 nvinfer1::TopKOperation topk_op; in ConvertArgMinMax() local
5690 topk_op = nvinfer1::TopKOperation::kMIN; in ConvertArgMinMax()
5692 topk_op = nvinfer1::TopKOperation::kMAX; in ConvertArgMinMax()
5713 *inputs.at(0).tensor(), topk_op, 1, reduce_axes); in ConvertArgMinMax()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 2089 "//tensorflow/core/kernels:topk_op",
|