Searched defs:topk (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sort_ops_test.py | 254 def topk(v, k=k): function
290 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
349 auto topk = CreateTopKUnique(builder, unique, input_shape, k_, true); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | topk_rewriter.cc | 233 HloInstruction* topk = comp->AddInstruction( in TransformToCustomCall() local
|
| D | buffer_assignment.cc | 783 absl::btree_multimap<int64_t, const HloValue*> topk; in TopKPeakBuffers() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | custom_call_handler.cc | 133 auto topk = b_.AddInstruction( in HandleCustomCallTopK() local
|
| D | spmd_partitioner.cc | 2501 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 | 7008 auto topk = ops::TopK(s.WithOpName("my_topk"), input, weights); in TEST_P() local
|