Home
last modified time | relevance | path

Searched refs:accumulator_type (Results 1 – 9 of 9) sorted by relevance

/external/Reactive-Extensions/RxCpp/Rx/v2/src/rxcpp/operators/
Drx-scan.hpp47 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()
Drx-reduce.hpp88 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/
Dsoftmax_op_gpu.cu.cc63 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/
Drx-predef.hpp312 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/
Dir_emitter.h400 const ShardedVectorType& accumulator_type, HloInstruction* init_value,
Dir_emitter.cc1533 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/
Delemental_ir_emitter.cc2729 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/
Dcuda_dnn.cc3053 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()
Dcuda_dnn.h627 DeviceMemory<OutputType>* output_data, dnn::DataType accumulator_type,