Searched refs:reduce_function (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 508 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument
515 acc = reduce_function(acc, matrix(i, j)); in ReduceToColArray2D()
525 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument
532 acc = reduce_function(acc, matrix(j, i)); in ReduceToRowArray2D()
541 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument
567 accumulator = reduce_function( in Reduce4DTo1D()
616 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
628 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()
|