Searched defs:accumulator_type (Results 1 – 6 of 6) sorted by relevance
63 using accumulator_type = T; typedef68 using accumulator_type = float; typedef
47 typedef rxu::decay_t<Accumulator> accumulator_type; typedef
88 typedef rxu::decay_t<Accumulator> accumulator_type; typedef104 typedef typename traits::accumulator_type accumulator_type; typedef
312 typedef rxu::decay_t<Accumulator> accumulator_type; typedef
1568 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction()1823 PrimitiveType accumulator_type = reduce->shape().element_type(); in EmitElementalReduce() local
2696 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve() local2932 DeviceMemory<ElementType>* output_data, dnn::DataType accumulator_type, in DoFusedConvolveImpl()