| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 702 ReduceType reduce_type) { in EvalType() argument
728 switch (reduce_type) { in EvalType()
762 TF_LITE_KERNEL_LOG(context, "Unsupported ReduceType: %d", reduce_type); in EvalType()
806 GetTensorData<int>(normalized_dims), reduce_type)); in EvalType()
813 template <KernelType kernel_type, ReduceType reduce_type>
819 reduce_type); in EvalGeneric()
823 reduce_type); in EvalGeneric()
827 reduce_type); in EvalGeneric()
831 reduce_type); in EvalGeneric()
835 reduce_type); in EvalGeneric()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 499 ReduceType reduce_type) { in ReduceDispatcher() argument
501 switch (reduce_type) { in ReduceDispatcher()
528 switch (reduce_type) { in ReduceDispatcher()
557 ReduceType reduce_type) {
559 switch (reduce_type) {
579 switch (reduce_type) {
784 ReduceType reduce_type) { in ReduceGeneric() argument
799 resolved_axis, num_resolved_axis, reduce_type); in ReduceGeneric()
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 203 def risc_reduce(x, axis, reduce_type, name='RISC_REDUCE'): argument
204 return gen_risc_ops.risc_reduce(x, axis, reduce_type=reduce_type, name=name)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 356 PrimitiveType reduce_type = new_filter->shape().element_type(); in HandleBatchGroupCount() local
361 if (primitive_util::BitWidth(reduce_type) < primitive_util::BitWidth(F32)) { in HandleBatchGroupCount()
362 reduce_type = F32; in HandleBatchGroupCount()
370 auto zero_literal = LiteralUtil::Zero(reduce_type); in HandleBatchGroupCount()
376 Shape shape = ShapeUtil::MakeShape(reduce_type, {}); in HandleBatchGroupCount()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RiscReduce.pbtxt | 16 name: "reduce_type"
|
| /external/gemmlowp/meta/generators/ |
| D | neon_emitter.py | 666 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
669 if reduce_type is not 'u32':
670 raise ArgumentError('Unsupported reduce: %s' % reduce_type)
693 self.EmitVPadd(reduce_type, destinations[i], sources[2 * i],
699 self.EmitVPadd(reduce_type, sources[2 * i], sources[2 * i],
|
| D | neon_emitter_64.py | 1057 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
1060 if reduce_type is not 'u32':
1061 raise ArgumentError('Unsupported reduce: %s' % reduce_type)
1082 self.EmitVPadd(reduce_type, destinations[i], sources[2 * i],
1088 self.EmitVPadd(reduce_type, sources[2 * i], sources[2 * i],
|
| /external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| D | collective_lowering.cc | 614 absl::string_view reduce_type = kReduceOpAdd; in LowerAllGatherOp() local
615 if (type && type.getElementType().isInteger(1)) reduce_type = kReduceOpAny; in LowerAllGatherOp()
620 builder.getStringAttr(std::string(reduce_type)), in LowerAllGatherOp()
|
| /external/pytorch/test/ |
| D | test_sparse_csr.py | 2621 def run_test(m, n, k, nnz, reduce_type, index_dtype, train): argument
2641 out.scatter_reduce_(0, index, src, reduce=reduce_type, include_self=False)
2651 out = torch.sparse.mm(csr, mat, reduce_type)
2667 for reduce_type in ["sum", "mean", "amax", "amin"]:
2669 run_test(3, 4, 11, 1, reduce_type, index_dtype, train)
2670 run_test(3, 4, 11, 6, reduce_type, index_dtype, train)
2671 run_test(3, 4, 11, 12, reduce_type, index_dtype, train)
2674 run_test(4, 7, 33, 13, reduce_type, index_dtype, train)
|
| D | test_mps.py | 7582 for reduce_type in ["add", "multiply"]:
7583 helper((2, 3), 0, (5, 3), (5, 3), reduce_str=reduce_type)
7584 helper((2, 8, 4, 5), 0, (10, 8, 4, 5), (10, 8, 4, 5), reduce_str=reduce_type)
7585 helper((8, 8, 4, 5), 0, (10, 8, 4, 5), (10, 8, 4, 5), reduce_str=reduce_type)
7586 helper((8, 8, 4, 5), 0, (4, 7, 3, 2), (4, 7, 3, 2), reduce_str=reduce_type)
7587 helper((8, 8, 4, 5), 0, (4, 6, 3, 2), (4, 7, 3, 2), reduce_str=reduce_type)
7588 helper((8, 8, 4, 5), 0, (4, 6, 3, 2), (8, 8, 4, 5), reduce_str=reduce_type)
7590 helper((2, 8, 4, 5), 1, (2, 20, 4, 5), (2, 20, 4, 5), reduce_str=reduce_type)
7591 helper((2, 8, 4, 5), 1, (2, 13, 3, 2), (2, 13, 3, 2), reduce_str=reduce_type)
7592 helper((8, 8, 4, 5), 1, (6, 5, 2, 3), (6, 5, 2, 3), reduce_str=reduce_type)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 2703 RankedTensorType reduce_type = in convertReduceOpCommon() local
2706 auto reduce_op = CreateOpAndInfer<T>(rewriter, op->getLoc(), reduce_type, in convertReduceOpCommon()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 45919 name: "reduce_type"
|