Home
last modified time | relevance | path

Searched refs:precision_proto (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc347 PrecisionConfig precision_proto; in InvertDiagonalBlocks() local
348 precision_proto.add_operand_precision(precision); in InvertDiagonalBlocks()
349 precision_proto.add_operand_precision(precision); in InvertDiagonalBlocks()
350 auto update = -DotGeneral(input_row, body_out, dnums, &precision_proto); in InvertDiagonalBlocks()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc511 PrecisionConfig precision_proto; in Einsum() local
512 precision_proto.add_operand_precision(precision); in Einsum()
513 precision_proto.add_operand_precision(precision); in Einsum()
514 auto dot = DotGeneral(x, y, dnums, &precision_proto); in Einsum()