Home
last modified time | relevance | path

Searched refs:precision_config_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_svd_op.cc35 precision_config_.ParsePartialFromString(precision_config_attr), in XlaSvdOp()
37 if (precision_config_.operand_precision_size() == 0) { in XlaSvdOp()
38 precision_config_.mutable_operand_precision()->Add( in XlaSvdOp()
44 precision_config_.operand_precision(0)); in Compile()
53 xla::PrecisionConfig precision_config_; member in tensorflow::__anon9078a52a0111::XlaSvdOp
Dxla_dot_op.cc43 precision_config_.ParsePartialFromString(precision_config_attr), in XlaDotOp()
56 &precision_config_, preferred_element_type_); in Compile()
65 xla::PrecisionConfig precision_config_; member in tensorflow::__anon170500040111::XlaDotOp
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h1628 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1629 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1661 PrecisionConfig precision_config_; variable
1879 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1880 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1952 PrecisionConfig precision_config_; variable
2264 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
2265 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
2291 PrecisionConfig precision_config_; variable
Dhlo_instructions.cc2441 precision_config_(precision_config) { in HloConvolutionInstruction()
2470 *proto.mutable_precision_config() = precision_config_; in ToProto()
2491 PrecisionConfigToString(precision_config_); in ExtraAttributesToStringImpl()
2526 precision_config_); in CloneWithNewOperandsImpl()
2724 *proto.mutable_precision_config() = precision_config_; in ToProto()
2769 PrecisionConfigToString(precision_config_); in ExtraAttributesToStringImpl()
3277 precision_config_(precision_config) { in HloDotInstruction()
3285 *proto.mutable_precision_config() = precision_config_; in ToProto()
3295 PrecisionConfigToString(precision_config_); in ExtraAttributesToStringImpl()
3319 precision_config_); in CloneWithNewOperandsImpl()