Home
last modified time | relevance | path

Searched defs:precision_config (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc209 const ConvOpAttrs& attrs, const xla::PrecisionConfig* precision_config) { in MakeXlaForwardConvOp()
297 const xla::PrecisionConfig* precision_config) { in MakeXlaBackpropInputConvOp()
376 const ConvOpAttrs& attrs, const xla::PrecisionConfig* precision_config) { in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py244 precision_config=None, argument
290 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): argument
306 def svd(a, max_iter, epsilon, precision_config=None): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc1034 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs, in ComputeArrayForDotWithIndexedLhs()
1069 const PrecisionConfig& precision_config, ConstantArray* lhs, in ComputeArrayForDotWithIndexedRhs()
1104 const PrecisionConfig& precision_config, Array* lhs, Array* rhs) { in ComputeArrayForDot()
Dhlo_computation_test.cc553 PrecisionConfig precision_config; in TEST_F() local
588 PrecisionConfig precision_config; in TEST_F() local
624 PrecisionConfig precision_config; in TEST_F() local
Dhlo_creation_utils.cc86 const PrecisionConfig& precision_config) { in MakeConvolveHlo()
266 const PrecisionConfig& precision_config) { in MakeDotHlo()
Dhlo_instructions.h1141 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1583 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
Dhlo_instruction.cc455 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
598 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
862 const PrecisionConfig& precision_config) { in CreateConvolve()
896 const PrecisionConfig& precision_config) { in CreateDot()
3506 const PrecisionConfig& HloInstruction::precision_config() const { in precision_config() function in xla::HloInstruction
Dbfloat16_normalization_test.cc412 PrecisionConfig precision_config; in TEST_F() local
Dhlo.proto193 xla.PrecisionConfig precision_config = 51; field
Dhlo_instructions.cc52 string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString()
1981 const PrecisionConfig& precision_config) in HloConvolutionInstruction()
2666 const PrecisionConfig& precision_config) in HloDotInstruction()
Dhlo_parser.cc1154 PrecisionConfig precision_config; in ParseInstructionRhs() local
1687 PrecisionConfig precision_config; in ParseInstructionRhs() local
Dhlo_dataflow_analysis_test.cc2602 PrecisionConfig precision_config; in TEST_F() local
Dhlo_evaluator.cc377 const PrecisionConfig& precision_config, const Literal& lhs, in EvaluateDotOp()
Dbuffer_assignment_test.cc1774 PrecisionConfig precision_config; in TEST_F() local
Dalgebraic_simplifier_test.cc3622 PrecisionConfig precision_config; in TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1135 const PrecisionConfig* precision_config) { in Dot()
1149 const PrecisionConfig* precision_config) { in DotGeneral()
1214 const PrecisionConfig* precision_config) { in Conv()
1225 const PrecisionConfig* precision_config) { in ConvWithGeneralPadding()
1235 const PrecisionConfig* precision_config) { in ConvWithGeneralDimensions()
1272 const PrecisionConfig* precision_config) { in ConvGeneral()
1284 const PrecisionConfig* precision_config) { in ConvGeneralDilated()
3024 const PrecisionConfig* precision_config) { in Dot()
3030 const PrecisionConfig* precision_config) { in DotGeneral()
3038 const PrecisionConfig* precision_config) { in Conv()
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1373 def DotGeneral(self, lhs, rhs, dimension_numbers, precision_config=None): argument
1397 precision_config=None): argument
1432 precision_config=None): argument
1485 precision_config=None): argument
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc707 PrecisionConfig precision_config; in CreateCanonicalDot() local
Dhlo_test_base.cc160 PrecisionConfig precision_config; in DefaultPrecisionConfig() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc43 PrecisionConfig precision_config; in MakeDot() local
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc476 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc220 auto precision_config = absl::make_unique<xla::PrecisionConfig>(); in Convert_precision_config() local