Searched refs:precision_proto (Results 1 – 2 of 2) sorted by relevance
248 PrecisionConfig precision_proto; in Einsum() local249 precision_proto.add_operand_precision(precision); in Einsum()250 precision_proto.add_operand_precision(precision); in Einsum()251 return Transpose(DotGeneral(x, y, dnums, &precision_proto), transpose_dims); in Einsum()
226 PrecisionConfig precision_proto; in InvertDiagonalBlocks() local227 precision_proto.add_operand_precision(precision); in InvertDiagonalBlocks()228 precision_proto.add_operand_precision(precision); in InvertDiagonalBlocks()229 auto update = -DotGeneral(input_row, body_out, dnums, &precision_proto); in InvertDiagonalBlocks()