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 | 63 using accumulator_type = T; typedef
68 using accumulator_type = float; typedef
221 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 | 400 const ShardedVectorType& accumulator_type, HloInstruction* init_value,
|
| D | ir_emitter.cc | 1533 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction() argument
1537 accumulator.reserve(accumulator_type.size()); in EmitInnerLoopForVectorizedReduction()
1538 for (auto accumulator_shard_type : accumulator_type) { in EmitInnerLoopForVectorizedReduction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2729 PrimitiveType accumulator_type = element_shape.element_type(); in EmitElementalReduce() local
2731 llvm_ir::PrimitiveTypeToIrType(accumulator_type, module_); in EmitElementalReduce()
2844 llvm::Type* accumulator_type = in EmitConvolution() local
2847 accumulator_type, "convolution_sum_address", b_); in EmitConvolution()
2848 llvm::Value* constant_zero = llvm::Constant::getNullValue(accumulator_type); in EmitConvolution()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3053 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve() local
3055 ToCudnnDataType(accumulator_type)); in DoConvolve()
3187 DeviceMemory<OutputType>* output_data, dnn::DataType accumulator_type, in DoFusedConvolveImpl() argument
3222 ToCudnnDataType(accumulator_type)); in DoFusedConvolveImpl()
|
| D | cuda_dnn.h | 627 DeviceMemory<OutputType>* output_data, dnn::DataType accumulator_type,
|