Searched defs:topk (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sort_ops_test.py | 264 def topk(v, k=k): function
301 def topk(v, k=k): function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | topk_rewriter.cc | 158 HloInstruction* topk = comp->AddInstruction( in Run() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | topk_ops.cc | 219 auto topk = CreateTopKUnique(builder, input, input_shape, k_, false); in Compile() local
350 auto topk = CreateTopKUnique(builder, unique, input_shape, k_, true); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1692 auto topk = b_.AddInstruction(hlo->CloneWithNewOperands( in HandleSort() local
1847 auto topk = b_.AddInstruction( in HandleCustomCall() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5208 auto topk = ops::TopK(s.WithOpName("my_topk"), input, weights); in TEST_P() local
|