Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc189 const ConvOpAttrs& attrs, const xla::PrecisionConfig* precision_config) { in MakeXlaForwardConvOp()
296 const xla::PrecisionConfig* precision_config, xla::XlaOp* input_sizes) { in MakeXlaBackpropInputConvOp()
394 const ConvOpAttrs& attrs, const xla::PrecisionConfig* precision_config) { in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py251 precision_config=None, argument
297 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): argument
313 def svd(a, max_iter, epsilon, precision_config=None): argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1286 const PrecisionConfig* precision_config, in Dot()
1301 const PrecisionConfig* precision_config, in DotGeneral()
1318 const PrecisionConfig* precision_config) { in DotGeneralInternal()
1376 const PrecisionConfig* precision_config, in Conv()
1389 const PrecisionConfig* precision_config, in ConvWithGeneralPadding()
1401 const PrecisionConfig* precision_config, in ConvWithGeneralDimensions()
1440 const PrecisionConfig* precision_config, in ConvGeneral()
1454 const PrecisionConfig* precision_config, in ConvGeneralDilated()
1492 const PrecisionConfig* precision_config, PaddingType padding_type, in DynamicConvInstruction()
1534 const PrecisionConfig* precision_config, PaddingType padding_type, in DynamicConvInputGrad()
[all …]
/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_instructions.h1259 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1476 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1789 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
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.cc97 const PrecisionConfig& precision_config, in MakeConvolveHlo()
289 const PrecisionConfig& precision_config, in MakeDotHlo()
Dhlo_instruction.cc512 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
601 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
703 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
987 const PrecisionConfig& precision_config) { in CreateConvolve()
1029 const PrecisionConfig& precision_config) { in CreateDot()
3852 const PrecisionConfig& HloInstruction::precision_config() const { in precision_config() function in xla::HloInstruction
Dbfloat16_normalization_test.cc413 PrecisionConfig precision_config; in TEST_F() local
Dhlo.proto197 xla.PrecisionConfig precision_config = 51; field
Dhlo_parser.cc1634 PrecisionConfig precision_config; in ParseInstructionRhs() local
2368 PrecisionConfig precision_config; in ParseInstructionRhs() local
2419 PrecisionConfig precision_config; in ParseInstructionRhs() local
Dhlo_instructions.cc54 string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString()
2137 const PrecisionConfig& precision_config) in HloConvolutionInstruction()
2950 const PrecisionConfig& precision_config) in HloDotInstruction()
Ddynamic_padder.cc851 PrecisionConfig precision_config; in RewriteDynamicConvolutionInputGrad() local
Dhlo_dataflow_analysis_test.cc2625 PrecisionConfig precision_config; in TEST_F() local
Dhlo_evaluator.cc384 const PrecisionConfig& precision_config, const Literal& lhs, in EvaluateDotOp()
Dbuffer_assignment_test.cc1767 PrecisionConfig precision_config; in TEST_F() local
Dalgebraic_simplifier.cc5377 PrecisionConfig precision_config; in SwapConvOperands() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc148 PrecisionConfig precision_config; in ShuffleWithinEachPartitionUsingOneHot() local
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc100 const PrecisionConfig* precision_config) { in ConvGeneralDilatedInternal()
380 const PrecisionConfig* precision_config) { in DotGeneralInternal()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc719 PrecisionConfig precision_config; in CreateCanonicalDot() local
Dhlo_test_base.cc162 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.cc477 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc218 auto precision_config = absl::make_unique<xla::PrecisionConfig>(); in Convert_precision_config() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc541 ArrayAttr precision_config, OpBuilder *builder) { in BatchDot()
5629 auto precision_config = rewriter.getStrArrayAttr({"HIGHEST", "HIGHEST"}); in matchAndRewrite() local

12