Home
last modified time | relevance | path

Searched refs:logical_input_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dapprox_topk_shape.cc59 uint64_t logical_input_size = in ApproxTopKReductionOutputSize() local
65 log2_ceil(CeilOfRatio(logical_input_size, tpu_tiling)); in ApproxTopKReductionOutputSize()
99 uint32_t log2_reduction = log2_floor(logical_input_size / m); in ApproxTopKReductionOutputSize()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc1410 uint64_t logical_input_size = reduction_input_size_override >= 0 in ApproxTopKShape() local
1419 uint32_t log2_reduction = log2_floor(logical_input_size / m); in ApproxTopKShape()