Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dquantize_and_dequantize_op.cc217 string round_mode_string; in QuantizeAndDequantizeV2Op() local
218 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeAndDequantizeV2Op()
221 (round_mode_string == "HALF_UP" || round_mode_string == "HALF_TO_EVEN"), in QuantizeAndDequantizeV2Op()
225 round_mode_string + "'")); in QuantizeAndDequantizeV2Op()
226 if (round_mode_string == "HALF_UP") { in QuantizeAndDequantizeV2Op()
228 } else if (round_mode_string == "HALF_TO_EVEN") { in QuantizeAndDequantizeV2Op()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_op.cc84 string round_mode_string; in QuantizeV2Op() local
85 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeV2Op()
87 (round_mode_string == "HALF_AWAY_FROM_ZERO" || in QuantizeV2Op()
88 round_mode_string == "HALF_TO_EVEN"), in QuantizeV2Op()
92 round_mode_string + "'")); in QuantizeV2Op()
93 if (round_mode_string == "HALF_AWAY_FROM_ZERO") { in QuantizeV2Op()
95 } else if (round_mode_string == "HALF_TO_EVEN") { in QuantizeV2Op()
H A Dquantize_and_dequantize_op.cc60 string round_mode_string; in QuantizeAndDequantizeV2Op() local
61 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeAndDequantizeV2Op()
64 (round_mode_string == "HALF_UP" || round_mode_string == "HALF_TO_EVEN"), in QuantizeAndDequantizeV2Op()
68 round_mode_string + "'")); in QuantizeAndDequantizeV2Op()
69 if (round_mode_string == "HALF_UP") { in QuantizeAndDequantizeV2Op()
71 } else if (round_mode_string == "HALF_TO_EVEN") { in QuantizeAndDequantizeV2Op()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_quantize_op.cc242 string round_mode_string; in MklQuantizeV2Op() local
243 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in MklQuantizeV2Op()
245 (round_mode_string == "HALF_AWAY_FROM_ZERO" || in MklQuantizeV2Op()
246 round_mode_string == "HALF_TO_EVEN"), in MklQuantizeV2Op()
250 round_mode_string + "'")); in MklQuantizeV2Op()
251 if (round_mode_string == "HALF_AWAY_FROM_ZERO") { in MklQuantizeV2Op()
253 } else if (round_mode_string == "HALF_TO_EVEN") { in MklQuantizeV2Op()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc1610 string round_mode_string; in QuantizeOpRewrite() local
1615 TF_CHECK_OK(GetNodeAttr(n->def(), "round_mode", &round_mode_string)); in QuantizeOpRewrite()
1631 if (!((mode_string == "SCALED" && round_mode_string == "HALF_TO_EVEN") || in QuantizeOpRewrite()