Home
last modified time | relevance | path

Searched refs:reduce_function (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.cc510 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument
517 acc = reduce_function(acc, matrix(i, j)); in ReduceToColArray2D()
527 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument
534 acc = reduce_function(acc, matrix(j, i)); in ReduceToRowArray2D()
543 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument
569 accumulator = reduce_function( in Reduce4DTo1D()
618 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
630 reduce_function((*result)(row, col), array(i0, i1, i2)); in Reduce3DTo2D()
H A Dreference_util.h116 const std::function<float(float, float)>& reduce_function);
123 const std::function<float(float, float)>& reduce_function);
146 const std::function<float(float, float)>& reduce_function);
157 const std::function<float(float, float)>& reduce_function);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_group_converter.cc374 auto reduce_function = [&]() -> HloComputation* { in HandleBatchGroupCount() local
404 reduce_function())); in HandleBatchGroupCount()
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtflite-schema-v2.15.0.fbs1462 reduce_function: ReduceWindowFunction;