Searched defs:gamma_min (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_batch_norm_op_test.cc91 const float gamma_min = 0.0f; local
96 FloatTensorToQuantized<quint8>(gamma_float, gamma_min, gamma_max);
116 AddInputFromArray<float>(TensorShape({1}), {gamma_min});
188 const float gamma_min = 0.0f; local
193 FloatTensorToQuantized<quint8>(gamma_float, gamma_min, gamma_max);
213 AddInputFromArray<float>(TensorShape({1}), {gamma_min});
H A Dquantized_batch_norm_op.cc36 float gamma_min, float gamma_max,
65 QuantizedToFloat(gamma_flat(channel), gamma_min, gamma_max);
99 float gamma_min, float gamma_max,
130 QuantizedToFloat(gamma_flat(channel), gamma_min, gamma_max);
188 const float gamma_min = context->input(13).flat<float>()(0); variable
214 beta_max, gamma, gamma_min, gamma_max,
31 ReferenceBatchNorm(const Tensor& input, const float input_min, const float input_max, const Tensor& mean, float mean_min, float mean_max, const Tensor& var, float var_min, float var_max, const Tensor& beta, float beta_min, float beta_max, const Tensor& gamma, float gamma_min, float gamma_max, float variance_epsilon, bool scale_after_normalization, Tensor* output, float* output_min, float* output_max) argument
94 FixedPointBatchNorm(const Tensor& input, const float input_min, const float input_max, const Tensor& mean, float mean_min, float mean_max, const Tensor& var, float var_min, float var_max, const Tensor& beta, float beta_min, float beta_max, const Tensor& gamma, float gamma_min, float gamma_max, float variance_epsilon, bool scale_after_normalization, Tensor* output, float* output_min, float* output_max) argument

Completed in 95 milliseconds