Home
last modified time | relevance | path

Searched refs:narrow_range_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_quantize_ops.cc123 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in FakeQuantWithMinMaxArgsOp()
124 quant_min_ = narrow_range_ ? 1 : 0; in FakeQuantWithMinMaxArgsOp()
145 num_bits_, narrow_range_)); in Compile()
163 bool narrow_range_; member in tensorflow::__anonce4ef7ee0111::FakeQuantWithMinMaxArgsOp
234 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in FakeQuantWithMinMaxVarsOp()
235 quant_min_ = narrow_range_ ? 1 : 0; in FakeQuantWithMinMaxVarsOp()
250 b, input, input_min, input_max, num_bits_, narrow_range_)); in Compile()
266 bool narrow_range_; member in tensorflow::__anonce4ef7ee0111::FakeQuantWithMinMaxVarsOp
Dquantize_and_dequantize_op.cc52 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in QuantizeAndDequantizeOp()
123 if (narrow_range_) { in Compile()
205 bool narrow_range_; member in tensorflow::__anon210cb6f00111::QuantizeAndDequantizeOp
/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op.cc69 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in QuantizeAndDequantizeV2Op()
118 narrow_range_, output->flat<T>()); in Compute()
124 round_mode_, narrow_range_, in Compute()
135 bool narrow_range_; member in tensorflow::QuantizeAndDequantizeV2Op
221 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in QuantizeAndDequantizeV3Op()
277 ROUND_HALF_TO_EVEN, narrow_range_, output->flat<T>()); in Compute()
283 ROUND_HALF_TO_EVEN, narrow_range_, in Compute()
292 bool narrow_range_; member in tensorflow::QuantizeAndDequantizeV3Op
Ddequantize_op.cc88 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in DequantizeOp()
176 std::numeric_limits<T>::min() + (narrow_range_ ? 1 : 0); in DequantizeTensor()
213 std::numeric_limits<T>::min() + (narrow_range_ ? 1 : 0); in DequantizeSlice()
226 bool narrow_range_; member in tensorflow::DequantizeOp
Dquantize_op.cc103 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in QuantizeV2Op()
264 std::numeric_limits<T>::min() + (narrow_range_ ? 1 : 0); in QuantizeSlice()
302 bool narrow_range_; member in tensorflow::QuantizeV2Op
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_quantize_op.cc252 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in MklQuantizeV2Op()
490 bool narrow_range_; member in tensorflow::MklQuantizeV2Op