Searched refs:reduce_function (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 509 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument
516 acc = reduce_function(acc, matrix(i, j)); in ReduceToColArray2D()
526 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument
533 acc = reduce_function(acc, matrix(j, i)); in ReduceToRowArray2D()
542 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument
568 accumulator = reduce_function( in Reduce4DTo1D()
617 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
629 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 | 369 auto reduce_function = [&]() -> HloComputation* { in HandleBatchGroupCount() local
399 reduce_function())); in HandleBatchGroupCount()
|