@Operator(group="quantization") public final class FakeQuantWithMinMaxVarsPerChannelGradient extends PrimitiveOp
Modifier and Type | Class and Description |
---|---|
static class |
FakeQuantWithMinMaxVarsPerChannelGradient.Options
Optional attributes for
FakeQuantWithMinMaxVarsPerChannelGradient |
operation
Modifier and Type | Method and Description |
---|---|
Output<Float> |
backpropsWrtInput()
Backpropagated gradients w.r.t.
|
Output<Float> |
backpropWrtMax()
Backpropagated gradients w.r.t.
|
Output<Float> |
backpropWrtMin()
Backpropagated gradients w.r.t.
|
static FakeQuantWithMinMaxVarsPerChannelGradient |
create(Scope scope,
Operand<Float> gradients,
Operand<Float> inputs,
Operand<Float> min,
Operand<Float> max,
FakeQuantWithMinMaxVarsPerChannelGradient.Options... options)
Factory method to create a class wrapping a new FakeQuantWithMinMaxVarsPerChannelGradient operation.
|
static FakeQuantWithMinMaxVarsPerChannelGradient.Options |
narrowRange(Boolean narrowRange) |
static FakeQuantWithMinMaxVarsPerChannelGradient.Options |
numBits(Long numBits) |
equals, hashCode, op, toString
public static FakeQuantWithMinMaxVarsPerChannelGradient create(Scope scope, Operand<Float> gradients, Operand<Float> inputs, Operand<Float> min, Operand<Float> max, FakeQuantWithMinMaxVarsPerChannelGradient.Options... options)
scope
- current scopegradients
- Backpropagated gradients above the FakeQuantWithMinMaxVars operation,
shape one of: `[d]`, `[b, d]`, `[b, h, w, d]`.inputs
- Values passed as inputs to the FakeQuantWithMinMaxVars operation, shape
same as `gradients`.
min, max: Quantization interval, floats of shape `[d]`.min
- max
- options
- carries optional attributes valuespublic static FakeQuantWithMinMaxVarsPerChannelGradient.Options numBits(Long numBits)
numBits
- The bitwidth of the quantization; between 2 and 16, inclusive.public static FakeQuantWithMinMaxVarsPerChannelGradient.Options narrowRange(Boolean narrowRange)
narrowRange
- Whether to quantize into 2^num_bits - 1 distinct values.public Output<Float> backpropsWrtInput()
public Output<Float> backpropWrtMin()
Copyright © 2022. All rights reserved.