Searched refs:float_mean (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_instance_norm.cc | 320 Eigen::Tensor<float, 2, Eigen::RowMajor> float_mean(N, C); in Compute() local
327 H * W, C, float_mean.data(), float_variance.data()); in Compute()
332 C, float_mean.data(), float_variance.data(), in Compute()
341 C, float_mean.data(), float_variance.data(), in Compute()
350 float_mean = input_tensor.cast<float>().reduce( in Compute()
356 float_mean.reshape(expand_spec).broadcast(broadcast_spec)))) in Compute()
363 float_mean.reshape(expand_spec).broadcast(broadcast_spec)) * in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 394 float float_mean = static_cast<float>(temp_sum[idx]) / in QuantizedMeanOrSum() local
397 TfLiteRound(float_mean * scale + bias) + output_zero_point, in QuantizedMeanOrSum()
|