Searched defs:accumulator_type (Results 1 – 5 of 5) sorted by relevance
63 using accumulator_type = T; typedef68 using accumulator_type = float; typedef
69 std::string accumulator_type; in GetMetalDefines() local
2930 PrimitiveType accumulator_type = element_shape.element_type(); in EmitElementalReduce() local3048 llvm::Type* accumulator_type = in EmitConvolution() local
3564 auto accumulator_type = ToCudnnDataType(GetConvAccumulatorType(input_type)); in GetCudnnOperationGraph() local3641 dnn::DataType accumulator_type = GetConvAccumulatorType(input_type); in GetCudnnFusedOperationGraph() local4276 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve() local
1629 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction()