Searched refs:input_max_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantize_and_dequantize_op.cc | 307 OP_REQUIRES_OK(ctx, ctx->GetAttr("input_max", &input_max_)); in QuantizeAndDequantizeOp()
310 ctx, input_min_ <= input_max_, in QuantizeAndDequantizeOp()
312 " > input_max ", input_max_)); in QuantizeAndDequantizeOp()
327 input_max_tensor.template scalar<T>()() = static_cast<T>(input_max_); in Compute()
340 float input_max_; member in tensorflow::QuantizeAndDequantizeOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 128 OP_REQUIRES_OK(ctx, ctx->GetAttr("max", &input_max_)); in FakeQuantWithMinMaxArgsOp()
129 CpuNudge(input_min_, input_max_, quant_min_, quant_max_, &nudged_input_min_, in FakeQuantWithMinMaxArgsOp()
144 XlaHelpers::FloatLiteral(b, DataType::DT_FLOAT, input_max_), in Compile()
165 float input_max_; member in tensorflow::__anonce4ef7ee0111::FakeQuantWithMinMaxArgsOp
|