| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk_shape.cc | 39 int64_t input_size, int64_t rank, int64_t top_k, float recall_target, in ApproxTopKReductionOutputSize() argument
71 if (recall_target == 1.0) { in ApproxTopKReductionOutputSize()
75 if (recall_target <= 0. || recall_target > 1.0) { in ApproxTopKReductionOutputSize()
96 std::log(static_cast<double>(recall_target))), in ApproxTopKReductionOutputSize()
|
| D | approx_topk.h | 56 float recall_target = 0.9, bool aggregate_to_topk = true,
64 float recall_target = 0.9,
|
| D | approx_topk.cc | 160 float recall_target, bool aggregate_to_topk, in ApproxTopK() argument
197 n, rank, top_k, recall_target, /*aggregate_to_topk=*/false, in ApproxTopK()
256 const XlaComputation& comparator, float recall_target, in ApproxTopKFallback() argument
264 n, rank, top_k, recall_target, aggregate_to_topk, in ApproxTopKFallback()
|
| D | approx_topk_shape.h | 43 int64_t input_size, int64_t rank, int64_t top_k, float recall_target,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | approx_topk_op.cc | 101 float recall_target, bool aggregate_to_topk,
124 float recall_target, bool aggregate_to_topk, in ApproxTopKFn() argument
127 comparator, recall_target, aggregate_to_topk, in ApproxTopKFn()
142 float recall_target, bool aggregate_to_topk, in ApproxTopKFn() argument
146 recall_target, aggregate_to_topk, reduction_input_size_override); in ApproxTopKFn()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ApproxTopK.pbtxt | 28 name: "recall_target"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ApproxTopK.pbtxt | 35 name: "recall_target"
|
| /external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| D | ops.pyi | 85 recall_target: Optional[float],
95 recall_target: Optional[float],
102 recall_target: float,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 25 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
29 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
|
| D | tensorflow.math.pbtxt | 37 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
41 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
|
| 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/python/ops/ |
| D | nn_ops.py | 5798 recall_target=0.95, argument
5849 recall_target=recall_target,
5861 recall_target=0.95, argument
5916 recall_target=recall_target,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.math.pbtxt | 37 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
41 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
|
| D | tensorflow.nn.pbtxt | 17 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
21 …argspec: "args=[\'operand\', \'k\', \'reduction_dimension\', \'recall_target\', \'reduction_input_…
|
| 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 | nn_ops.cc | 1376 float recall_target; in ApproxTopKShape() local
1381 TF_RETURN_IF_ERROR(c->GetAttr("recall_target", &recall_target)); in ApproxTopKShape()
1404 if (reduction_dim_value <= tpu_tiling || recall_target == 1.0) { in ApproxTopKShape()
1416 std::log(static_cast<double>(recall_target))), in ApproxTopKShape()
|
| D | ops.pbtxt | 2253 name: "recall_target"
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1610 recall_target = 0.95
1639 recall_target=recall_target)
1652 self.assertGreater(hits / (qy_size * k), recall_target)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 535 DefaultValuedAttr<F32Attr, "0.95f">:$recall_target,
|