Searched refs:accum_type (Results 1 – 4 of 4) sorted by relevance
110 accum_type(hlo->to_apply()->root_instruction()->shape().element_type()), in GroupKey()121 if (accum_type != other.accum_type) { in operator <()122 return accum_type < other.accum_type; in operator <()152 PrimitiveType accum_type; member
65 std::string accum_type; in GenerateConvolutionTransposedCode() local70 accum_type = "float" + type_postfix; in GenerateConvolutionTransposedCode()73 accum_type = "half" + type_postfix; in GenerateConvolutionTransposedCode()91 c += " " + accum_type + " r[" + std::to_string(kernel_size.y) + "][" + in GenerateConvolutionTransposedCode()
262 const DataType accum_type = op_def.precision == CalculationsPrecision::F16 in GetWinograd4x4To36TileX6Code() local277 std::string cl_type = accum_type == DataType::FLOAT16 ? "half" : "float"; in GetWinograd4x4To36TileX6Code()543 const DataType accum_type = op_def.precision == CalculationsPrecision::F16 in GetWinograd36To4x4Tile4x1Code() local547 std::string cl_type = accum_type == DataType::FLOAT16 ? "half" : "float"; in GetWinograd36To4x4Tile4x1Code()
615 llvm::Type* accum_type = target_array_.GetElementLlvmType(); in EmitNaiveLlvmIrGemm() local617 b_->CreateAlloca(accum_type, /*ArraySize=*/nullptr, "accum_address"); in EmitNaiveLlvmIrGemm()624 b_->CreateStore(llvm::Constant::getNullValue(accum_type), accum_address); in EmitNaiveLlvmIrGemm()