Searched refs:round_mode_string (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | quantize_and_dequantize_op.cc | 217 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 D | quantize_op.cc | 84 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 D | quantize_and_dequantize_op.cc | 60 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 D | mkl_quantize_op.cc | 242 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 D | mkl_layout_pass.cc | 1610 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()
|