| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 188 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/ |
| D | xla.py | 254 precision_config=None, argument
323 precision_config=None, argument
353 def svd(a, max_iter, epsilon, precision_config=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1297 const PrecisionConfig* precision_config, in Dot()
1313 const PrecisionConfig* precision_config, in DotGeneral()
1330 const PrecisionConfig* precision_config) { in DotGeneralInternal()
1388 const PrecisionConfig* precision_config, in Conv()
1401 const PrecisionConfig* precision_config, in ConvWithGeneralPadding()
1413 const PrecisionConfig* precision_config, in ConvWithGeneralDimensions()
1452 const PrecisionConfig* precision_config, in ConvGeneral()
1466 const PrecisionConfig* precision_config, in ConvGeneralDilated()
1504 const PrecisionConfig* precision_config, PaddingType padding_type, in DynamicConvInstruction()
1546 const PrecisionConfig* precision_config, PaddingType padding_type, in DynamicConvInputGrad()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lower_general_dot.cc | 157 ArrayAttr precision_config; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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 | 1328 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1561 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
1890 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config() function
|
| D | hlo_creation_utils.cc | 97 const PrecisionConfig& precision_config, in MakeConvolveHlo()
289 const PrecisionConfig& precision_config, in MakeDotHlo()
|
| D | hlo_instruction.cc | 625 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
714 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
818 PrecisionConfig precision_config = proto.precision_config(); in CreateFromProto() local
1104 const PrecisionConfig& precision_config) { in CreateConvolve()
1146 const PrecisionConfig& precision_config) { in CreateDot()
4149 const PrecisionConfig& HloInstruction::precision_config() const { in precision_config() function in xla::HloInstruction
|
| D | bfloat16_normalization_test.cc | 413 PrecisionConfig precision_config; in TEST_F() local
|
| D | hlo.proto | 232 xla.PrecisionConfig precision_config = 51; field
|
| D | hlo_parser.cc | 1771 PrecisionConfig precision_config; in ParseInstructionRhs() local
2524 PrecisionConfig precision_config; in ParseInstructionRhs() local
2575 PrecisionConfig precision_config; in ParseInstructionRhs() local
|
| D | hlo_instructions.cc | 56 string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString()
2275 const PrecisionConfig& precision_config) in HloConvolutionInstruction()
3115 const PrecisionConfig& precision_config) in HloDotInstruction()
|
| D | dynamic_padder.cc | 950 PrecisionConfig precision_config; in RewriteDynamicConvolutionInputGrad() local
|
| D | hlo_evaluator.cc | 419 const PrecisionConfig& precision_config, const Literal& lhs, in EvaluateDotOp()
|
| D | hlo_dataflow_analysis_test.cc | 2703 PrecisionConfig precision_config; in TEST_F() local
|
| D | buffer_assignment_test.cc | 1806 PrecisionConfig precision_config; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 148 PrecisionConfig precision_config; in ShuffleWithinEachPartitionUsingOneHot() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 100 const PrecisionConfig* precision_config) { in ConvGeneralDilatedInternal()
397 const PrecisionConfig* precision_config) { in DotGeneralInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 721 PrecisionConfig precision_config; in CreateCanonicalDot() local
|
| D | hlo_test_base.cc | 162 PrecisionConfig precision_config; in DefaultPrecisionConfig() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 43 PrecisionConfig precision_config; in MakeDot() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.cc | 478 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 236 auto precision_config = absl::make_unique<xla::PrecisionConfig>(); in Convert_precision_config() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 491 ArrayAttr precision_config, OpBuilder *builder) { in BatchDot()
6732 auto precision_config = rewriter.getStrArrayAttr({"HIGHEST", "HIGHEST"}); in matchAndRewrite() local
|