Searched refs:accumulator_type (Results 1 – 9 of 9) 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 | 67 using accumulator_type = T; typedef
72 using accumulator_type = float; typedef
159 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.cc | 1035 llvm::Type* accumulator_type = in EmitElementalConvolution() local
1038 accumulator_type, "convolution_sum_address", &b_, in EmitElementalConvolution()
1040 llvm::Value* constant_zero = llvm::Constant::getNullValue(accumulator_type); in EmitElementalConvolution()
1149 llvm::Value* sum = FAdd(Load(sum_address), FPCast(product, accumulator_type)); in EmitElementalConvolution()
1756 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction() argument
1760 accumulator.reserve(accumulator_type.size()); in EmitInnerLoopForVectorizedReduction()
1761 for (auto accumulator_shard_type : accumulator_type) { in EmitInnerLoopForVectorizedReduction()
2037 PrimitiveType accumulator_type = element_shape.element_type(); in EmitElementalReduce() local
2039 llvm_ir::PrimitiveTypeToIrType(accumulator_type, module_); in EmitElementalReduce()
2045 MinimumAlignmentForPrimitiveType(accumulator_type)); in EmitElementalReduce()
|
| D | ir_emitter.h | 412 const ShardedVectorType& accumulator_type, HloInstruction* init_value,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 725 PrimitiveType accumulator_type = element_shape.element_type(); in HandleReduce() local
727 llvm_ir::PrimitiveTypeToIrType(accumulator_type, module_); in HandleReduce()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2966 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve() local
2968 ToCudnnDataType(accumulator_type)); in DoConvolve()
3223 DeviceMemory<OutputType>* output_data, dnn::DataType accumulator_type, in DoFusedConvolveImpl() argument
3245 ToCudnnDataType(accumulator_type)); in DoFusedConvolveImpl()
|
| D | cuda_dnn.h | 634 DeviceMemory<OutputType>* output_data, dnn::DataType accumulator_type,
|