Searched refs:input_scale (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
H A D | quantization_util.h | 53 void PreprocessSoftmaxScaling(double beta, double input_scale,
|
H A D | quantization_util.cc | 62 void PreprocessSoftmaxScaling(double beta, double input_scale, argument 75 beta * input_scale * (1 << (31 - input_integer_bits)), (1ll << 31) - 1.0);
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | fake_quantize_ops.cc | 76 const xla::ComputationDataHandle& input_scale) { 78 xla::ComputationDataHandle inv_scale = b->Div(one, input_scale); 88 return b->Add(b->Mul(rounded, input_scale), nudged_input_min); 122 xla::ComputationDataHandle input_scale = variable 125 b, input, data_type, nudged_input_min, nudged_input_max, input_scale); 213 xla::ComputationDataHandle nudged_input_min, nudged_input_max, input_scale; variable 215 &nudged_input_min, &nudged_input_max, &input_scale); 218 b, input, data_type, nudged_input_min, nudged_input_max, input_scale); 254 xla::ComputationDataHandle nudged_input_min, nudged_input_max, input_scale; variable 256 &nudged_input_min, &nudged_input_max, &input_scale); 71 Quantize( xla::ComputationBuilder* b, const xla::ComputationDataHandle& input, const DataType data_type, const xla::ComputationDataHandle& nudged_input_min, const xla::ComputationDataHandle& nudged_input_max, const xla::ComputationDataHandle& input_scale) argument [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
H A D | quantized_instance_norm.cc | 279 float input_scale = (input_max - input_min) / 255.0f; variable 354 (input_scale * 361 input_scale *
|
Completed in 611 milliseconds