Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.h94 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT,
98 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT,
129 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT,
133 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT);
144 xla::PrecisionConfig::Precision precision = xla::PrecisionConfig::DEFAULT,
Dqr_test.cc65 xla::BatchDot(q, r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
102 xla::BatchDot(q, r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
132 xla::BatchDot(q, r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
153 xla::BatchDot(q, r, xla::PrecisionConfig::HIGHEST); in XLA_TEST_F()
Dsvd.h45 PrecisionConfig::Precision precision = PrecisionConfig::HIGHEST);
Dmatrix.cc386 xla::PrecisionConfig::Precision precision, in Einsum()
544 PrecisionConfig precision_proto; in Einsum()
573 XlaOp BatchDot(XlaOp x, XlaOp y, PrecisionConfig::Precision precision, in BatchDot()
579 PrecisionConfig::Precision precision, in BatchDot()
710 PrecisionConfig::Precision precision, in Einsum()
729 PrecisionConfig::Precision precision) { in Einsum()
Dsvd_test.cc96 PrecisionConfig::HIGHEST); in ComputeMatmulUDVT()
168 BatchDot(result.v, TransposeInMinorDims(result.v), PrecisionConfig::HIGHEST); in XLA_TEST_F()
Dself_adjoint_eig_test.cc131 PrecisionConfig::HIGHEST); in ComputeMatmulVWVt()
197 BatchDot(result.v, TransposeInMinorDims(result.v), PrecisionConfig::HIGHEST); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dqr_expander.h37 XlaOp a, PrecisionConfig::Precision precision);
41 XlaOp taus, int64_t m, int64_t n, PrecisionConfig::Precision precision);
45 PrecisionConfig::Precision precision);
49 PrecisionConfig::Precision precision);
Dtriangular_solve_expander.h47 PrecisionConfig::Precision precision);
51 PrecisionConfig::Precision precision);
57 PrecisionConfig::Precision precision);
62 PrecisionConfig::Precision precision);
Dcholesky_expander.h36 XlaOp a, PrecisionConfig::Precision precision);
40 PrecisionConfig::Precision precision);
Dhlo_module_config.h341 PrecisionConfig::Precision matrix_unit_operand_precision() const { in matrix_unit_operand_precision()
345 PrecisionConfig::Precision matrix_unit_operand_precision) { in set_matrix_unit_operand_precision()
452 PrecisionConfig::Precision matrix_unit_operand_precision_ =
453 PrecisionConfig::DEFAULT;
Dtriangular_solve_expander.cc131 PrecisionConfig::Precision precision) { in SolveWithInvertedDiagonalBlocks()
240 PrecisionConfig::Precision precision) { in InvertDiagonalBlocks()
340 PrecisionConfig precision_proto; in InvertDiagonalBlocks()
368 PrecisionConfig::Precision precision) { in SolveByInvertingDiagonalBlocks()
412 PrecisionConfig::Precision precision) { in SolveDirectly()
469 PrecisionConfig::Precision precision) { in BuildTriangularSolve()
593 /*precision=*/PrecisionConfig::HIGHEST); in ExpandInstruction()
Dqr_expander.cc200 XlaOp a, PrecisionConfig::Precision precision) { in QrBlock()
320 XlaOp taus, int64_t m, int64_t n, PrecisionConfig::Precision precision) { in CompactWYRepresentation()
374 XlaOp a, int64_t block_size, PrecisionConfig::Precision precision) { in BuildQrDecomposition()
438 PrecisionConfig::Precision precision) { in ProductOfElementaryHouseholderReflectors()
542 /*precision=*/PrecisionConfig::HIGHEST)); in ExpandInstruction()
549 /*precision=*/PrecisionConfig::HIGHEST)); in ExpandInstruction()
Dcholesky_expander.cc56 XlaOp a, PrecisionConfig::Precision precision) { in CholeskyUnblocked()
127 PrecisionConfig::Precision precision) { in BuildCholesky()
246 /*precision=*/PrecisionConfig::HIGHEST); in ExpandInstruction()
Dindexed_array_analysis.h271 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs,
276 const PrecisionConfig& precision_config, ConstantArray* lhs,
281 const PrecisionConfig& precision_config,
Dhlo_computation_test.cc535 PrecisionConfig precision_config; in TEST_F()
537 2, PrecisionConfig::DEFAULT); in TEST_F()
571 PrecisionConfig precision_config; in TEST_F()
573 2, PrecisionConfig::DEFAULT); in TEST_F()
608 PrecisionConfig precision_config; in TEST_F()
610 2, PrecisionConfig::DEFAULT); in TEST_F()
Dhlo_instructions.h1599 const PrecisionConfig& precision_config);
1628 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1629 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1661 PrecisionConfig precision_config_;
1879 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1880 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1952 PrecisionConfig precision_config_;
2250 const PrecisionConfig& precision_config);
2264 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
2265 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
[all …]
/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/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.cc39 xla::PrecisionConfig::HIGHEST); in XlaSvdOp()
53 xla::PrecisionConfig precision_config_;
Dbatch_matmul_op.cc47 xla::PrecisionConfig::DEFAULT, preferred_element_type_); in Compile()
Dmatrix_solve_op.cc61 /*transpose_y=*/false, xla::PrecisionConfig::HIGHEST); in Compile()
Dxla_dot_op.cc65 xla::PrecisionConfig precision_config_;
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h526 const PrecisionConfig* precision_config = nullptr,
531 const PrecisionConfig* precision_config = nullptr,
538 const PrecisionConfig* precision_config = nullptr,
545 const PrecisionConfig* precision_config = nullptr,
552 const PrecisionConfig* precision_config = nullptr,
560 const PrecisionConfig* precision_config = nullptr,
570 const PrecisionConfig* precision_config = nullptr,
581 const PrecisionConfig* precision_config, PaddingType padding_type,
592 const PrecisionConfig* precision_config, PaddingType padding_type,
603 const PrecisionConfig* precision_config, PaddingType padding_type,
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.h31 mlir::ArrayAttr ConvertPrecisionConfig(const PrecisionConfig* config,
/external/tensorflow/tensorflow/compiler/xla/python/
Dtypes.h419 struct type_caster<xla::PrecisionConfig> {
421 PYBIND11_TYPE_CASTER(xla::PrecisionConfig, _("xla::PrecisionConfig"));
434 operand_precision.cast<xla::PrecisionConfig::Precision>());

123