Searched refs:reduce_function (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 573 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument
580 acc = reduce_function(acc, matrix(i, j)); in ReduceToColArray2D()
590 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument
597 acc = reduce_function(acc, matrix(j, i)); in ReduceToRowArray2D()
606 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument
632 accumulator = reduce_function( in Reduce4DTo1D()
681 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
693 reduce_function((*result)(row, col), array(i0, i1, i2)); in Reduce3DTo2D()
|
| D | reference_util.h | 118 const std::function<float(float, float)>& reduce_function);
125 const std::function<float(float, float)>& reduce_function);
148 const std::function<float(float, float)>& reduce_function);
159 const std::function<float(float, float)>& reduce_function);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 272 auto reduce_function = [&]() -> HloComputation* { in HandleBatchGroupCount() local
302 reduce_function())); in HandleBatchGroupCount()
|