Searched defs:reduce_type (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 499 ReduceType reduce_type) { in ReduceDispatcher()
784 ReduceType reduce_type) { in ReduceGeneric()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 356 PrimitiveType reduce_type = new_filter->shape().element_type(); in HandleBatchGroupCount() local
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 203 def risc_reduce(x, axis, reduce_type, name='RISC_REDUCE'): argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| D | collective_lowering.cc | 614 absl::string_view reduce_type = kReduceOpAdd; in LowerAllGatherOp() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 702 ReduceType reduce_type) { in EvalType()
|
| /external/gemmlowp/meta/generators/ |
| D | neon_emitter.py | 666 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
|
| D | neon_emitter_64.py | 1057 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 2703 RankedTensorType reduce_type = in convertReduceOpCommon() local
|