Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
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()
/external/tensorflow/tensorflow/core/kernels/
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()
Dquantize_and_dequantize_op.cc55 string round_mode_string; in QuantizeAndDequantizeV2Op() local
56 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeAndDequantizeV2Op()
59 (round_mode_string == "HALF_UP" || round_mode_string == "HALF_TO_EVEN"), in QuantizeAndDequantizeV2Op()
63 round_mode_string + "'")); in QuantizeAndDequantizeV2Op()
64 if (round_mode_string == "HALF_UP") { in QuantizeAndDequantizeV2Op()
66 } else if (round_mode_string == "HALF_TO_EVEN") { in QuantizeAndDequantizeV2Op()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_quantize_op.cc233 string round_mode_string; in MklQuantizeV2Op() local
234 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in MklQuantizeV2Op()
236 (round_mode_string == "HALF_AWAY_FROM_ZERO" || in MklQuantizeV2Op()
237 round_mode_string == "HALF_TO_EVEN"), in MklQuantizeV2Op()
241 round_mode_string + "'")); in MklQuantizeV2Op()
242 if (round_mode_string == "HALF_AWAY_FROM_ZERO") { in MklQuantizeV2Op()
244 } else if (round_mode_string == "HALF_TO_EVEN") { in MklQuantizeV2Op()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1658 string round_mode_string; in QuantizeOpRewrite() local
1663 TF_CHECK_OK(GetNodeAttr(n->def(), "round_mode", &round_mode_string)); in QuantizeOpRewrite()
1679 if (!((mode_string == "SCALED" && round_mode_string == "HALF_TO_EVEN") || in QuantizeOpRewrite()