| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 236 precision_config=None, argument
282 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): argument
298 def svd(a, max_iter, epsilon, precision_config=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1011 const PrecisionConfig* precision_config) { in Dot()
1025 const PrecisionConfig* precision_config) { in DotGeneral()
1102 const PrecisionConfig* precision_config) { in Conv()
1113 const PrecisionConfig* precision_config) { in ConvWithGeneralPadding()
1123 const PrecisionConfig* precision_config) { in ConvWithGeneralDimensions()
1160 const PrecisionConfig* precision_config) { in ConvGeneral()
1172 const PrecisionConfig* precision_config) { in ConvGeneralDilated()
2959 const PrecisionConfig* precision_config) { in Dot()
2965 const PrecisionConfig* precision_config) { in DotGeneral()
2973 const PrecisionConfig* precision_config) { in Conv()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 85 const PrecisionConfig& precision_config) { in MakeConvolveHlo()
219 const PrecisionConfig& precision_config) { in MakeDotHlo()
|
| D | indexed_array_analysis.cc | 1034 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()
|
| D | hlo_computation_test.cc | 553 PrecisionConfig precision_config; in TEST_F() local
588 PrecisionConfig precision_config; in TEST_F() local
624 PrecisionConfig precision_config; in TEST_F() local
|
| D | hlo_instructions.h | 1060 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1481 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
|
| D | bfloat16_normalization_test.cc | 353 PrecisionConfig precision_config; in TEST_F() local
|
| D | hlo_instruction.cc | 418 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
558 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
784 const PrecisionConfig& precision_config) { in CreateConvolve()
818 const PrecisionConfig& precision_config) { in CreateDot()
3176 const PrecisionConfig& HloInstruction::precision_config() const { in precision_config() function in xla::HloInstruction
|
| D | hlo.proto | 188 xla.PrecisionConfig precision_config = 51; field
|
| D | hlo_instructions.cc | 51 string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString()
1858 const PrecisionConfig& precision_config) in HloConvolutionInstruction()
2512 const PrecisionConfig& precision_config) in HloDotInstruction()
|
| D | hlo_parser.cc | 1092 PrecisionConfig precision_config; in ParseInstructionRhs() local
1601 PrecisionConfig precision_config; in ParseInstructionRhs() local
|
| D | tuple_points_to_analysis_test.cc | 1129 PrecisionConfig precision_config; in TEST_F() local
|
| D | hlo_evaluator.cc | 365 const PrecisionConfig& precision_config, const Literal& lhs, in EvaluateDotOp()
|
| D | hlo_dataflow_analysis_test.cc | 2508 PrecisionConfig precision_config; in TEST_F() local
|
| D | buffer_assignment_test.cc | 1595 PrecisionConfig precision_config; in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 3236 PrecisionConfig precision_config; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 580 PrecisionConfig precision_config; in CreateCanonicalDot() local
|
| D | hlo_test_base.cc | 175 PrecisionConfig precision_config; in DefaultPrecisionConfig() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 42 PrecisionConfig precision_config; in MakeDot() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 542 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
|