Searched refs:accumulator_type (Results 1 – 8 of 8) sorted by relevance
| /external/Reactive-Extensions/RxCpp/Rx/v2/src/rxcpp/operators/ |
| D | rx-scan.hpp | 47 typedef rxu::decay_t<Accumulator> accumulator_type; typedef
52 scan_initial_type(source_type o, accumulator_type a, seed_type s) in scan_initial_type()
59 accumulator_type accumulator;
64 scan(source_type o, accumulator_type a, seed_type s) in scan()
|
| D | rx-reduce.hpp | 88 typedef rxu::decay_t<Accumulator> accumulator_type; typedef
104 typedef typename traits::accumulator_type accumulator_type; typedef
116 reduce_initial_type(source_type o, accumulator_type a, result_selector_type rs, seed_type s) in reduce_initial_type()
124 accumulator_type accumulator;
136 reduce(source_type o, accumulator_type a, result_selector_type rs, seed_type s) in reduce()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | softmax_op_gpu.cu.cc | 63 using accumulator_type = T; typedef
68 using accumulator_type = float; typedef
155 typedef typename softmax_traits<T>::accumulator_type acc_type; in Compute()
|
| /external/Reactive-Extensions/RxCpp/Rx/v2/src/rxcpp/ |
| D | rx-predef.hpp | 312 typedef rxu::decay_t<Accumulator> accumulator_type; typedef
322 typedef decltype(check<seed_type, source_value_type, accumulator_type>(0)) type;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 399 const ShardedVectorType& accumulator_type, HloInstruction* init_value,
|
| D | ir_emitter.cc | 1568 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction() argument
1572 accumulator.reserve(accumulator_type.size()); in EmitInnerLoopForVectorizedReduction()
1573 for (auto accumulator_shard_type : accumulator_type) { in EmitInnerLoopForVectorizedReduction()
1823 PrimitiveType accumulator_type = reduce->shape().element_type(); in EmitElementalReduce() local
1825 llvm_ir::PrimitiveTypeToIrType(accumulator_type, module_), "accumulator", in EmitElementalReduce()
1826 &b_, MinimumAlignmentForPrimitiveType(accumulator_type)); in EmitElementalReduce()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 593 DeviceMemory<ElementType>* output_data, dnn::DataType accumulator_type,
|
| D | cuda_dnn.cc | 2696 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve() local
2698 ToCudnnDataType(accumulator_type)); in DoConvolve()
2932 DeviceMemory<ElementType>* output_data, dnn::DataType accumulator_type, in DoFusedConvolveImpl() argument
2954 ToCudnnDataType(accumulator_type)); in DoFusedConvolveImpl()
|