Searched refs:is_max_k (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | approx_topk_op.cc | 37 bool is_max_k) { in ComparatorBuilder() argument
44 if (is_max_k) { in ComparatorBuilder()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ApproxTopK.pbtxt | 35 name: "is_max_k"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ApproxTopK.pbtxt | 39 name: "is_max_k"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 5850 is_max_k=True,
5917 is_max_k=False,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 453 …argspec: "args=[\'input\', \'k\', \'reduction_dimension\', \'recall_target\', \'is_max_k\', \'redu…
|
| D | tensorflow.raw_ops.pbtxt | 209 …argspec: "args=[\'input\', \'k\', \'reduction_dimension\', \'recall_target\', \'is_max_k\', \'redu…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 773 …argspec: "args=[\'input\', \'k\', \'reduction_dimension\', \'recall_target\', \'is_max_k\', \'redu…
|
| D | tensorflow.raw_ops.pbtxt | 209 …argspec: "args=[\'input\', \'k\', \'reduction_dimension\', \'recall_target\', \'is_max_k\', \'redu…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 2260 name: "is_max_k"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 536 DefaultValuedAttr<BoolAttr, "true">:$is_max_k,
|