| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | topk_rewriter.cc | 158 HloInstruction* topk = comp->AddInstruction( in Run() local
162 topk->shape().tuple_shapes(0), topk, 0)); in Run()
165 topk->shape().tuple_shapes(1), topk, 1)); in Run()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sort_ops_test.py | 264 def topk(v, k=k): function
268 topk, [x.astype(dtype)],
301 def topk(v, k=k): function
305 topk, [x.astype(dtype)],
319 topk = nn_ops.top_k(p, k=4)
321 topk,
337 topk = nn_ops.top_k(p, k=6)
338 results = sess.run(topk, {
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | topk_ops.cc | 219 auto topk = CreateTopKUnique(builder, input, input_shape, k_, false); in Compile() local
220 ctx->SetOutput(0, topk.first); in Compile()
221 ctx->SetOutput(1, topk.second); in Compile()
350 auto topk = CreateTopKUnique(builder, unique, input_shape, k_, true); in Compile() local
351 ctx->SetOutput(0, topk.first); in Compile()
352 ctx->SetOutput(1, topk.second); in Compile()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TopKUnique.pbtxt | 8 name: "topk"
|
| D | TopKWithUnique.pbtxt | 8 name: "topk"
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1692 auto topk = b_.AddInstruction(hlo->CloneWithNewOperands( in HandleSort() local
1698 topk->shape().tuple_shapes(0), topk, 0)); in HandleSort()
1701 topk->shape().tuple_shapes(1), topk, 1)); in HandleSort()
1847 auto topk = b_.AddInstruction( in HandleCustomCall() local
1849 topk->set_sharding(custom_call_sharding); in HandleCustomCall()
1851 PartitionedHlo partitioned_topk(topk, replicated_shape, in HandleCustomCall()
1853 topk = partitioned_topk.hlo(); in HandleCustomCall()
1858 topk->shape().tuple_shapes(0), topk, 0)); in HandleCustomCall()
1871 topk->shape().tuple_shapes(1), topk, 1)); in HandleCustomCall()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 144 from tensorflow.lite.testing.op_tests.topk import make_topk_tests
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 404 "topk",
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | ops.mlir | 963 // CHECK-LABEL: topk
964 func @topk(%arg0: tensor<8xf32>, %arg1: tensor<i32>) -> (tensor<?xf32>, tensor<?xi32>) {
971 // CHECK-LABEL: topk
972 func @topk(%arg0: tensor<*xf32>, %arg1: tensor<i32>) -> (tensor<*xf32>, tensor<*xi32>) {
|
| D | legalize-tf.mlir | 611 func @topk(%arg0: tensor<8xf32>, %arg1: tensor<i32>) -> (tensor<?xf32>, tensor<?xi32>) {
615 // CHECK-LABEL: topk
|
| /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
5209 const NodeDef& node_def = topk.operation.node()->def(); in TEST_P()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 20200 func TopKWithUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output…
34990 func TopKUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 56665 name: "topk"
56730 name: "topk"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 16878 TF_Float32Tensor:$topk,
16937 TF_Float32Tensor:$topk,
|