Home
last modified time | relevance | path

Searched refs:reduce_type (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc579 OpContext* op_context, ReduceType reduce_type) { in EvalType() argument
580 switch (reduce_type) { in EvalType()
613 OpContext* op_context, ReduceType reduce_type) { in EvalType() argument
614 switch (reduce_type) { in EvalType()
628 template <KernelType kernel_type, ReduceType reduce_type>
636 return EvalType<float>(context, node, &op_context, reduce_type); in EvalGeneric()
639 return EvalType<int>(context, node, &op_context, reduce_type); in EvalGeneric()
642 return EvalType<int64_t>(context, node, &op_context, reduce_type); in EvalGeneric()
645 return EvalType<uint8_t>(context, node, &op_context, reduce_type); in EvalGeneric()
648 return EvalType<int8_t>(context, node, &op_context, reduce_type); in EvalGeneric()
[all …]
/external/tensorflow/tensorflow/python/ops/risc/
Drisc_ops.py207 def risc_reduce(x, axis, reduce_type, name='RISC_REDUCE'): argument
208 return gen_risc_ops.risc_reduce(x, axis, reduce_type=reduce_type, name=name)
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc351 PrimitiveType reduce_type = new_filter->shape().element_type(); in HandleBatchGroupCount() local
356 if (primitive_util::BitWidth(reduce_type) < primitive_util::BitWidth(F32)) { in HandleBatchGroupCount()
357 reduce_type = F32; in HandleBatchGroupCount()
365 auto zero_literal = LiteralUtil::Zero(reduce_type); in HandleBatchGroupCount()
371 Shape shape = ShapeUtil::MakeShape(reduce_type, {}); in HandleBatchGroupCount()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DRiscReduce.pbtxt16 name: "reduce_type"
/external/gemmlowp/meta/generators/
Dneon_emitter.py666 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],
Dneon_emitter_64.py1057 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/compiler/mlir/tosa/transforms/
Dlegalize_common.cc2342 RankedTensorType reduce_type = RankedTensorType::get( in convertReduceOpCommon() local
2346 rewriter.create<T>(op->getLoc(), reduce_type, val, axis_attr); in convertReduceOpCommon()
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt42815 name: "reduce_type"