Home
last modified time | relevance | path

Searched refs:PrecisionConfig (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.h85 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT);
88 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT);
118 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT);
121 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT);
132 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT);
Dqr_test.cc60 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
89 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
121 xla::BatchDot(result.q, result.r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
Dqr.h38 PrecisionConfig::Precision precision = PrecisionConfig::HIGHEST);
Dsvd.h45 PrecisionConfig::Precision precision = PrecisionConfig::HIGHEST);
Dmatrix.cc355 xla::PrecisionConfig::Precision precision) { in Einsum()
511 PrecisionConfig precision_proto; in Einsum()
539 XlaOp BatchDot(XlaOp x, XlaOp y, PrecisionConfig::Precision precision) { in BatchDot()
544 PrecisionConfig::Precision precision) { in BatchDot()
674 PrecisionConfig::Precision precision) { in Einsum()
692 PrecisionConfig::Precision precision) { in Einsum()
Dqr.cc57 PrecisionConfig::Precision precision) { in QRDecomposition()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.h44 PrecisionConfig::Precision precision);
48 PrecisionConfig::Precision precision);
54 PrecisionConfig::Precision precision);
59 PrecisionConfig::Precision precision);
Dqr_expander.h45 PrecisionConfig::Precision precision);
49 XlaOp taus, int64 m, int64 n, PrecisionConfig::Precision precision);
53 PrecisionConfig::Precision precision);
Dcholesky_expander.h36 XlaOp a, PrecisionConfig::Precision precision);
40 PrecisionConfig::Precision precision);
Dtriangular_solve_expander.cc137 PrecisionConfig::Precision precision) { in SolveWithInvertedDiagonalBlocks()
246 PrecisionConfig::Precision precision) { in InvertDiagonalBlocks()
347 PrecisionConfig precision_proto; in InvertDiagonalBlocks()
375 PrecisionConfig::Precision precision) { in SolveByInvertingDiagonalBlocks()
419 PrecisionConfig::Precision precision) { in SolveDirectly()
476 PrecisionConfig::Precision precision) { in BuildTriangularSolve()
599 /*precision=*/PrecisionConfig::HIGHEST); in ExpandInstruction()
Dindexed_array_analysis.h270 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs,
275 const PrecisionConfig& precision_config, ConstantArray* lhs,
280 const PrecisionConfig& precision_config,
Dcholesky_expander.cc56 XlaOp a, PrecisionConfig::Precision precision) { in CholeskyUnblocked()
129 PrecisionConfig::Precision precision) { in BuildCholesky()
248 /*precision=*/PrecisionConfig::HIGHEST); in ExpandInstruction()
Dqr_expander.cc176 XlaOp a, PrecisionConfig::Precision precision) { in QrBlock()
296 XlaOp taus, int64 m, int64 n, PrecisionConfig::Precision precision) { in CompactWYRepresentation()
350 XlaOp a, int64 block_size, PrecisionConfig::Precision precision) { in BuildQrDecomposition()
448 /*precision=*/PrecisionConfig::HIGHEST)); in ExpandInstruction()
Dhlo_computation_test.cc553 PrecisionConfig precision_config; in TEST_F()
555 2, PrecisionConfig::DEFAULT); in TEST_F()
588 PrecisionConfig precision_config; in TEST_F()
590 2, PrecisionConfig::DEFAULT); in TEST_F()
624 PrecisionConfig precision_config; in TEST_F()
626 2, PrecisionConfig::DEFAULT); in TEST_F()
Dhlo_creation_utils.h68 const PrecisionConfig& precision_config,
140 const PrecisionConfig& precision_config,
Dhlo_instructions.h1230 const PrecisionConfig& precision_config);
1259 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1260 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1288 PrecisionConfig precision_config_;
1476 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1477 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1530 PrecisionConfig precision_config_;
1775 const PrecisionConfig& precision_config);
1789 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1790 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.h63 const xla::PrecisionConfig* precision_config = nullptr);
67 const xla::PrecisionConfig* precision_config = nullptr,
73 const xla::PrecisionConfig* precision_config = nullptr);
Dxla_svd_op.cc38 xla::PrecisionConfig::HIGHEST); in XlaSvdOp()
52 xla::PrecisionConfig precision_config_;
Dxla_dot_op.cc57 xla::PrecisionConfig precision_config_;
Dmatrix_solve_op.cc61 /*transpose_y=*/false, xla::PrecisionConfig::HIGHEST); in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcholesky_test.cc239 BatchDot(input, TransposeInMinorDims(input), PrecisionConfig::HIGHEST); in XLA_TEST_P()
247 PrecisionConfig::HIGHEST); in XLA_TEST_P()
250 PrecisionConfig::HIGHEST); in XLA_TEST_P()
284 PrecisionConfig::HIGHEST); in XLA_TEST_P()
292 PrecisionConfig::HIGHEST); in XLA_TEST_P()
295 PrecisionConfig::HIGHEST); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h525 XlaOp lhs, XlaOp rhs, const PrecisionConfig* precision_config = nullptr,
530 const PrecisionConfig* precision_config = nullptr,
537 const PrecisionConfig* precision_config = nullptr,
544 const PrecisionConfig* precision_config = nullptr,
551 const PrecisionConfig* precision_config = nullptr,
559 const PrecisionConfig* precision_config = nullptr,
569 const PrecisionConfig* precision_config = nullptr,
579 const PrecisionConfig* precision_config, PaddingType padding_type,
590 const PrecisionConfig* precision_config, PaddingType padding_type,
601 const PrecisionConfig* precision_config, PaddingType padding_type,
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.h29 mlir::ArrayAttr ConvertPrecisionConfig(const PrecisionConfig* config,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc558 py::enum_<PrecisionConfig::Precision>(m, "PrecisionConfig_Precision") in BuildXlaCompilerSubmodule()
559 .value("DEFAULT", PrecisionConfig::DEFAULT) in BuildXlaCompilerSubmodule()
560 .value("HIGH", PrecisionConfig::HIGH) in BuildXlaCompilerSubmodule()
561 .value("HIGHEST", PrecisionConfig::HIGHEST); in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py126 PRECISION_VALUES = (None, xla_data_pb2.PrecisionConfig.DEFAULT,
127 xla_data_pb2.PrecisionConfig.HIGH,
128 xla_data_pb2.PrecisionConfig.HIGHEST)
149 precision_config = xla_data_pb2.PrecisionConfig()
180 precision_config = xla_data_pb2.PrecisionConfig()

123