Searched refs:reduce_type (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | reduce.cc | 579 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/ |
D | risc_ops.py | 207 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/ |
D | convolution_group_converter.cc | 351 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/ |
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/compiler/mlir/tosa/transforms/ |
D | legalize_common.cc | 2342 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/ |
D | ops.pbtxt | 42815 name: "reduce_type"
|