Home
last modified time | relevance | path

Searched refs:precision_config (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py236 precision_config=None, argument
260 if precision_config:
261 precision_config_proto = precision_config.SerializeToString()
271 precision_config=precision_config_proto,
282 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): argument
284 if precision_config:
285 precision_config_proto = precision_config.SerializeToString()
290 precision_config=precision_config_proto,
298 def svd(a, max_iter, epsilon, precision_config=None): argument
300 if precision_config:
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py122 precision_config = None
124 precision_config = xla_data_pb2.PrecisionConfig()
125 precision_config.operand_precision.extend([precision, precision])
153 precision_config = None
155 precision_config = xla_data_pb2.PrecisionConfig()
156 precision_config.operand_precision.extend([precision, precision])
161 precision_config=precision_config)
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h377 const PrecisionConfig* precision_config = nullptr);
381 const PrecisionConfig* precision_config = nullptr);
386 const PrecisionConfig* precision_config = nullptr);
393 const PrecisionConfig* precision_config = nullptr);
400 const PrecisionConfig* precision_config = nullptr);
407 const PrecisionConfig* precision_config = nullptr);
417 const PrecisionConfig* precision_config = nullptr);
777 const PrecisionConfig* precision_config);
780 const PrecisionConfig* precision_config);
784 const PrecisionConfig* precision_config);
[all …]
Dxla_builder.cc1011 const PrecisionConfig* precision_config) { in Dot() argument
1019 return DotGeneral(lhs, rhs, dimension_numbers, precision_config); in Dot()
1025 const PrecisionConfig* precision_config) { in DotGeneral() argument
1047 if (precision_config != nullptr) { in DotGeneral()
1048 *instr.mutable_precision_config() = *precision_config; in DotGeneral()
1102 const PrecisionConfig* precision_config) { in Conv() argument
1106 feature_group_count, batch_group_count, precision_config); in Conv()
1113 const PrecisionConfig* precision_config) { in ConvWithGeneralPadding() argument
1116 feature_group_count, batch_group_count, precision_config); in ConvWithGeneralPadding()
1123 const PrecisionConfig* precision_config) { in ConvWithGeneralDimensions() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_computation_test.cc553 PrecisionConfig precision_config; in TEST_F() local
554 precision_config.mutable_operand_precision()->Resize( in TEST_F()
557 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
588 PrecisionConfig precision_config; in TEST_F() local
589 precision_config.mutable_operand_precision()->Resize( in TEST_F()
592 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
624 PrecisionConfig precision_config; in TEST_F() local
625 precision_config.mutable_operand_precision()->Resize( in TEST_F()
628 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config)); in TEST_F()
Dindexed_array_analysis.h270 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs,
275 const PrecisionConfig& precision_config, ConstantArray* lhs,
280 const PrecisionConfig& precision_config,
Dindexed_array_analysis.cc168 instr->precision_config(), in ComputeArrayFor()
1034 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs, in ComputeArrayForDotWithIndexedLhs() argument
1053 new_dim_numbers, precision_config, lhs->literal(), *rhs->literal()))); in ComputeArrayForDotWithIndexedLhs()
1069 const PrecisionConfig& precision_config, ConstantArray* lhs, in ComputeArrayForDotWithIndexedRhs() argument
1089 new_dim_numbers, precision_config, *lhs->literal(), rhs->literal()))); in ComputeArrayForDotWithIndexedRhs()
1104 const PrecisionConfig& precision_config, Array* lhs, Array* rhs) { in ComputeArrayForDot() argument
1127 precision_config, in ComputeArrayForDot()
1136 precision_config, lhs_constant, in ComputeArrayForDot()
Dhlo_creation_utils.h60 const PrecisionConfig& precision_config);
110 const PrecisionConfig& precision_config);
Dconvolution_group_converter.cc243 convolution->window(), dim_numbers, convolution->precision_config())); in HandleBatchGroupCount()
383 convolution->window(), dim_numbers, convolution->precision_config()); in HandleConvolution()
445 convolution->precision_config())); in HandleConvolution()
524 convolution->precision_config())); in HandleConvolution()
Dtranspose_folding.cc111 dot->shape(), new_lhs, new_rhs, new_dim_numbers, dot->precision_config()); in FoldTransposeIntoDot()
180 convolution.precision_config()); in FoldTransposeIntoConvolution()
Dhlo_creation_utils.cc85 const PrecisionConfig& precision_config) { in MakeConvolveHlo() argument
94 dimension_numbers, precision_config)); in MakeConvolveHlo()
219 const PrecisionConfig& precision_config) { in MakeDotHlo() argument
226 dot_shape, lhs, rhs, dim_numbers, precision_config)); in MakeDotHlo()
Dbfloat16_normalization_test.cc353 PrecisionConfig precision_config; in TEST_F() local
354 precision_config.mutable_operand_precision()->Resize( in TEST_F()
357 HloInstruction::CreateDot(bf16_shape, a, b, dot_dnums, precision_config)); in TEST_F()
Dbatch_dot_simplification.cc74 batch_dot->precision_config())); in ElideDegenerateBatchDimensionFromBatchDot()
Dhlo_instruction.cc418 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
419 precision_config.mutable_operand_precision()->Resize( in CreateFromProto()
425 proto.convolution_dimension_numbers(), precision_config); in CreateFromProto()
558 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
559 precision_config.mutable_operand_precision()->Resize( in CreateFromProto()
563 precision_config); in CreateFromProto()
784 const PrecisionConfig& precision_config) { in CreateConvolve() argument
787 dimension_numbers, precision_config); in CreateConvolve()
818 const PrecisionConfig& precision_config) { in CreateDot() argument
820 shape, lhs, rhs, dimension_numbers, precision_config); in CreateDot()
[all …]
Dhlo_instructions.cc51 string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString() argument
52 if (absl::c_all_of(precision_config.operand_precision(), [](int32 precision) { in PrecisionConfigToString()
62 precision_config.operand_precision(), ",", in PrecisionConfigToString()
1858 const PrecisionConfig& precision_config) in HloConvolutionInstruction() argument
1864 precision_config_(precision_config) { in HloConvolutionInstruction()
1937 protobuf_util::ProtobufEquals(precision_config(), in IdenticalSlowPath()
1938 casted_other.precision_config()); in IdenticalSlowPath()
2512 const PrecisionConfig& precision_config) in HloDotInstruction() argument
2515 precision_config_(precision_config) { in HloDotInstruction()
2545 protobuf_util::ProtobufEquals(precision_config(), in IdenticalSlowPath()
[all …]
Ddot_decomposer.cc141 dot_dnums, dot->precision_config())); in DecomposeBatchDot()
279 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config())); in CanonicalizeDot()
Dhlo_instructions.h1035 const PrecisionConfig& precision_config);
1060 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1467 const PrecisionConfig& precision_config);
1481 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
Dhlo_instruction.h441 const PrecisionConfig& precision_config);
465 const PrecisionConfig& precision_config);
1341 const PrecisionConfig& precision_config() const;
Dhlo_evaluator.h126 const PrecisionConfig& precision_config,
Dhlo_parser.cc1092 PrecisionConfig precision_config; in ParseInstructionRhs() local
1094 *precision_config.mutable_operand_precision() = { in ParseInstructionRhs()
1097 precision_config.mutable_operand_precision()->Resize( in ParseInstructionRhs()
1103 *dnums, precision_config)); in ParseInstructionRhs()
1601 PrecisionConfig precision_config; in ParseInstructionRhs() local
1603 *precision_config.mutable_operand_precision() = { in ParseInstructionRhs()
1606 precision_config.mutable_operand_precision()->Resize( in ParseInstructionRhs()
1611 shape, operands[0], operands[1], dnum, precision_config)); in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc175 PrecisionConfig precision_config; in DefaultPrecisionConfig() local
176 precision_config.mutable_operand_precision()->Resize( in DefaultPrecisionConfig()
178 return precision_config; in DefaultPrecisionConfig()
Dtest_utils.cc580 PrecisionConfig precision_config; in CreateCanonicalDot() local
581 precision_config.mutable_operand_precision()->Resize( in CreateCanonicalDot()
587 shape, lhs, rhs, dot_dimension_numbers, precision_config); in CreateCanonicalDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc136 hlo->precision_config())); in Run()
Dcpu_instruction_fusion_test.cc42 PrecisionConfig precision_config; in MakeDot() local
43 precision_config.mutable_operand_precision()->Resize( in MakeDot()
46 precision_config); in MakeDot()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc542 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
543 precision_config.mutable_operand_precision()->Resize( in ConvArray4DGeneralDimensionsDilated()
547 /*batch_group_count=*/1, window, dnums, precision_config)); in ConvArray4DGeneralDimensionsDilated()

12