Searched defs:aggregate_to_topk (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | approx_topk_op.cc | 124 float recall_target, bool aggregate_to_topk, in ApproxTopKFn()
142 float recall_target, bool aggregate_to_topk, in ApproxTopKFn()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | approx_topk_test.py | 71 def test_non_fused_max_k(self, k, row_size, num_rows, aggregate_to_topk): argument
94 def test_non_fused_min_k(self, k, row_size, num_rows, aggregate_to_topk): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | approx_topk_test.py | 74 aggregate_to_topk): argument
97 aggregate_to_topk): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | approx_topk.cc | 160 float recall_target, bool aggregate_to_topk, in ApproxTopK()
257 bool aggregate_to_topk, in ApproxTopKFallback()
|
| H A D | approx_topk_shape.cc | 40 bool aggregate_to_topk, int64_t input_size_override) { in ApproxTopKReductionOutputSize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 5800 aggregate_to_topk=True, argument
5863 aggregate_to_topk=True, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | nn_ops.cc | 1378 bool aggregate_to_topk; in ApproxTopKShape() local
|