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,
|