Home
last modified time | relevance | path

Searched defs:log2_reduction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dapprox_topk_shape.cc64 uint32_t log2_reduction = in ApproxTopKReductionOutputSize() local
99 uint32_t log2_reduction = log2_floor(logical_input_size / m); in ApproxTopKReductionOutputSize() local
H A Dapprox_topk.cc203 int64_t approx_output_size, log2_reduction; in ApproxTopK() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc1419 uint32_t log2_reduction = log2_floor(logical_input_size / m); in ApproxTopKShape() local