Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc442 OpContext* op_context, ReduceType reduce_type) { in EvalType() argument
443 switch (reduce_type) { in EvalType()
476 OpContext* op_context, ReduceType reduce_type) { in EvalType() argument
477 switch (reduce_type) { in EvalType()
491 template <KernelType kernel_type, ReduceType reduce_type>
499 return EvalType<float>(context, node, &op_context, reduce_type); in EvalGeneric()
502 return EvalType<int>(context, node, &op_context, reduce_type); in EvalGeneric()
505 return EvalType<int64_t>(context, node, &op_context, reduce_type); in EvalGeneric()
508 return EvalType<uint8_t>(context, node, &op_context, reduce_type); in EvalGeneric()
511 return EvalType<int8_t>(context, node, &op_context, reduce_type); in EvalGeneric()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc254 PrimitiveType reduce_type = new_filter->shape().element_type(); in HandleBatchGroupCount() local
259 if (primitive_util::BitWidth(reduce_type) < primitive_util::BitWidth(F32)) { in HandleBatchGroupCount()
260 reduce_type = F32; in HandleBatchGroupCount()
268 auto zero_literal = LiteralUtil::Zero(reduce_type); in HandleBatchGroupCount()
274 Shape shape = ShapeUtil::MakeShape(reduce_type, {}); in HandleBatchGroupCount()
/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],