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