Searched defs:topk (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sort_ops_test.py | 262 def topk(v, k=k): function
298 def topk(v, k=k): function
|
| /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/ |
| D | topk_rewriter.cc | 223 HloInstruction* topk = comp->AddInstruction( in TransformToCustomCall() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | custom_call_handler.cc | 131 auto topk = b_.AddInstruction( in HandleCustomCallTopK() local
|
| D | spmd_partitioner.cc | 1809 auto topk = b_.AddInstruction(hlo->CloneWithNewOperands( in HandleSort() local
|
| /external/python/google-api-python-client/googleapiclient/discovery_cache/documents/ |
| D | apigee.v1.json | 4857 "topk": { object
5554 "topk": { object
6093 "topk": { object
6822 "topk": { object
8643 "topk": { object
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5659 auto topk = ops::TopK(s.WithOpName("my_topk"), input, weights); in TEST_P() local
|