From a09cb5e2be064e6aae44add1d55fbbe158dfe89e Mon Sep 17 00:00:00 2001 From: shugeo Date: Thu, 10 Oct 2019 17:13:33 +0300 Subject: [PATCH] Added doc for fake_quant_with_min_max_per_channel op declaration. --- .../include/ops/declarable/headers/parity_ops.h | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/libnd4j/include/ops/declarable/headers/parity_ops.h b/libnd4j/include/ops/declarable/headers/parity_ops.h index e0dc55937..cbc7e56da 100644 --- a/libnd4j/include/ops/declarable/headers/parity_ops.h +++ b/libnd4j/include/ops/declarable/headers/parity_ops.h @@ -1747,6 +1747,22 @@ namespace nd4j { #if NOT_EXCLUDED(OP_fake_quant_with_min_max_vars) DECLARE_CONFIGURABLE_OP(fake_quant_with_min_max_vars, 3, 1, true, 0, -2); #endif + +/** + * fake_quant_with_min_max_vals_per_channel - tf.quantization.fake_quant_with_min_max_vars_per_channel + * + * input params: + * 0 - NDArray (input) - at least 2D. + * 1 - 1D Tensor - min values (min length equals to last dim of input) + * 2 - 1D Tensor - max value (length equals to min) + * + * int params (optional): + * 0 - num_bits (allowed interval [2, 16], default 8) + * 1 - narrow_range (default False) + * + * output: + * 0 - NDArray with the same shape as input + */ #if NOT_EXCLUDED(OP_fake_quant_with_min_max_vars_per_channel) DECLARE_CONFIGURABLE_OP(fake_quant_with_min_max_vars_per_channel, 3, 1, true, 0, -2); #endif