Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc1359 const PrecisionConfig* precision_config, in Dot()
1375 const PrecisionConfig* precision_config, in DotGeneral()
1392 const PrecisionConfig* precision_config) { in DotGeneralInternal()
1449 const PrecisionConfig* precision_config, in Conv()
1462 const PrecisionConfig* precision_config, in ConvWithGeneralPadding()
1474 const PrecisionConfig* precision_config, in ConvWithGeneralDimensions()
1513 const PrecisionConfig* precision_config, in ConvGeneral()
1531 const PrecisionConfig* precision_config, in ConvGeneralDilated()
1571 const PrecisionConfig* precision_config, PaddingType padding_type, in DynamicConvInstruction()
1615 const PrecisionConfig* precision_config, PaddingType padding_type, in DynamicConvInputGrad()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dconv_op_helpers.cc188 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/
H A Dxla.py251 precision_config=None, argument
329 precision_config=None, argument
359 def svd(a, max_iter, epsilon, precision_config=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dindexed_array_analysis.cc1038 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs, in ComputeArrayForDotWithIndexedLhs()
1073 const PrecisionConfig& precision_config, ConstantArray* lhs, in ComputeArrayForDotWithIndexedRhs()
1108 const PrecisionConfig& precision_config, Array* lhs, Array* rhs) { in ComputeArrayForDot()
H A Dhlo_computation_test.cc535 PrecisionConfig precision_config; in TEST_F() local
571 PrecisionConfig precision_config; in TEST_F() local
608 PrecisionConfig precision_config; in TEST_F() local
H A Dhlo_instructions.h1628 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1879 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
2264 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
H A Dhlo_instruction.cc681 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
772 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
882 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
1174 const PrecisionConfig& precision_config) { in CreateConvolve()
1243 const PrecisionConfig& precision_config) { in CreateDot()
4390 const PrecisionConfig& HloInstruction::precision_config() const { in precision_config() function in xla::HloInstruction
H A Dhlo_creation_utils.cc121 const PrecisionConfig& precision_config, in MakeConvolveHlo()
337 const PrecisionConfig& precision_config, in MakeDotHlo()
H A Dbfloat16_normalization_test.cc413 PrecisionConfig precision_config; in TEST_F() local
H A Dhlo_parser.cc2011 PrecisionConfig precision_config; in CreateInstruction() local
2733 PrecisionConfig precision_config; in CreateInstruction() local
2785 PrecisionConfig precision_config; in CreateInstruction() local
H A Dhlo.proto252 xla.PrecisionConfig precision_config = 51; field
H A Dhlo_instructions.cc66 std::string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString()
2435 const PrecisionConfig& precision_config) in HloConvolutionInstruction()
3274 const PrecisionConfig& precision_config) in HloDotInstruction()
H A Ddynamic_padder.cc1008 PrecisionConfig precision_config; in RewriteDynamicConvolutionInputGrad() local
H A Dhlo_dataflow_analysis_test.cc2686 PrecisionConfig precision_config; in TEST_F() local
H A Dbuffer_assignment_test.cc1803 PrecisionConfig precision_config; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.cc350 auto precision_config = op.getPrecisionConfig(); in For() local
617 auto precision_config = op.getPrecisionConfig(); in For() local
H A Dbackend_configs.proto59 xla.PrecisionConfig precision_config = 12; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc108 const PrecisionConfig* precision_config) { in ConvGeneralDilatedInternal()
501 const PrecisionConfig* precision_config) { in DotGeneralInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc146 PrecisionConfig precision_config; in ShuffleWithinEachPartitionUsingOneHot() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtest_utils.cc776 PrecisionConfig precision_config; in CreateCanonicalDot() local
H A Dhlo_test_base.cc184 PrecisionConfig precision_config; in DefaultPrecisionConfig() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc43 PrecisionConfig precision_config; in MakeDot() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.cc480 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc443 ArrayAttr precision_config, OpBuilder *builder) { in BatchDot()
6546 auto precision_config = rewriter.getArrayAttr( in matchAndRewrite() local
7044 xla::PrecisionConfig precision_config; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc329 auto precision_config = std::make_unique<xla::PrecisionConfig>(); in Convert_precision_config() local

12