Searched refs:topk (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sort_ops_test.py | 95 def topk(v, k=k): function
99 topk, [x.astype(dtype)],
123 def topk(v, k=k): function
127 topk, [x.astype(dtype)],
140 topk = nn_ops.top_k(p, k=4)
142 topk,
158 topk = nn_ops.top_k(p, k=6)
159 results = sess.run(topk, {
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 325 "topk",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3385 auto topk = ops::TopK(s.WithOpName("my_topk"), input, weights); in TEST_F() local
3386 const NodeDef& node_def = topk.operation.node()->def(); in TEST_F()
|