Searched refs:input_scale (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
H A Dquantization_util.h53 void PreprocessSoftmaxScaling(double beta, double input_scale,
H A Dquantization_util.cc62 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 Dfake_quantize_ops.cc76 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 Dquantized_instance_norm.cc279 float input_scale = (input_max - input_min) / 255.0f; variable
354 (input_scale *
361 input_scale *

Completed in 611 milliseconds