Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_svd_op.cc34 precision_config_.ParsePartialFromString(precision_config_attr), in XlaSvdOp()
36 if (precision_config_.operand_precision_size() == 0) { in XlaSvdOp()
37 precision_config_.mutable_operand_precision()->Add( in XlaSvdOp()
43 precision_config_.operand_precision(0)); in Compile()
52 xla::PrecisionConfig precision_config_; member in tensorflow::__anon8179f0250111::XlaSvdOp
Dxla_dot_op.cc40 precision_config_.ParsePartialFromString(precision_config_attr), in XlaDotOp()
51 dnums_, &precision_config_); in Compile()
57 xla::PrecisionConfig precision_config_; member in tensorflow::__anon08064aff0111::XlaDotOp
Dxla_conv_op.cc39 precision_config_.ParsePartialFromString(precision_config_attr), in XlaConvOp()
80 /*batch_group_count=*/1, &precision_config_); in Compile()
86 xla::PrecisionConfig precision_config_; member in tensorflow::__anon40743f2e0111::XlaConvOp
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h1060 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1061 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1090 PrecisionConfig precision_config_; variable
1481 const PrecisionConfig& precision_config() const { return precision_config_; } in precision_config()
1482 PrecisionConfig* mutable_precision_config() { return &precision_config_; } in mutable_precision_config()
1506 PrecisionConfig precision_config_; variable
Dhlo_instructions.cc1864 precision_config_(precision_config) { in HloConvolutionInstruction()
1893 *proto.mutable_precision_config() = precision_config_; in ToProto()
1913 string precision_config_string = PrecisionConfigToString(precision_config_); in ExtraAttributesToStringImpl()
1949 precision_config_); in CloneWithNewOperandsImpl()
2515 precision_config_(precision_config) { in HloDotInstruction()
2523 *proto.mutable_precision_config() = precision_config_; in ToProto()
2531 string precision_config_string = PrecisionConfigToString(precision_config_); in ExtraAttributesToStringImpl()
2555 precision_config_); in CloneWithNewOperandsImpl()