| /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/xla/service/gpu/ |
| D | backend_configs.proto | 59 xla.PrecisionConfig precision_config = 12; field
|
| D | matmul_utils.cc | 350 auto precision_config = op.getPrecisionConfig(); in For() local
617 auto precision_config = op.getPrecisionConfig(); in For() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 251 precision_config=None, argument
329 precision_config=None, argument
359 def svd(a, max_iter, epsilon, precision_config=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1359 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 …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 1038 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()
|
| D | hlo_computation_test.cc | 535 PrecisionConfig precision_config; in TEST_F() local
571 PrecisionConfig precision_config; in TEST_F() local
608 PrecisionConfig precision_config; in TEST_F() local
|
| D | hlo_instructions.h | 1628 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
|
| D | hlo_creation_utils.cc | 121 const PrecisionConfig& precision_config, in MakeConvolveHlo()
337 const PrecisionConfig& precision_config, in MakeDotHlo()
|
| D | bfloat16_normalization_test.cc | 413 PrecisionConfig precision_config; in TEST_F() local
|
| D | hlo_instruction.cc | 681 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
|
| D | hlo.proto | 252 xla.PrecisionConfig precision_config = 51; field
|
| D | hlo_parser.cc | 2011 PrecisionConfig precision_config; in CreateInstruction() local
2733 PrecisionConfig precision_config; in CreateInstruction() local
2785 PrecisionConfig precision_config; in CreateInstruction() local
|
| D | hlo_instructions.cc | 66 std::string PrecisionConfigToString(const PrecisionConfig& precision_config) { in PrecisionConfigToString()
2435 const PrecisionConfig& precision_config) in HloConvolutionInstruction()
3274 const PrecisionConfig& precision_config) in HloDotInstruction()
|
| D | dynamic_padder.cc | 1008 PrecisionConfig precision_config; in RewriteDynamicConvolutionInputGrad() local
|
| D | hlo_dataflow_analysis_test.cc | 2686 PrecisionConfig precision_config; in TEST_F() local
|
| D | buffer_assignment_test.cc | 1803 PrecisionConfig precision_config; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 146 PrecisionConfig precision_config; in ShuffleWithinEachPartitionUsingOneHot() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 108 const PrecisionConfig* precision_config) { in ConvGeneralDilatedInternal()
501 const PrecisionConfig* precision_config) { in DotGeneralInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 776 PrecisionConfig precision_config; in CreateCanonicalDot() local
|
| D | hlo_test_base.cc | 184 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 | 480 PrecisionConfig precision_config; in ConvArray4DGeneralDimensionsDilated() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 443 ArrayAttr precision_config, OpBuilder *builder) { in BatchDot()
6546 auto precision_config = rewriter.getArrayAttr( in matchAndRewrite() local
7044 xla::PrecisionConfig precision_config; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 329 auto precision_config = std::make_unique<xla::PrecisionConfig>(); in Convert_precision_config() local
|