HomeSort by relevance Sort by last modified time
    Searched refs:mean_flat (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
quantized_batch_norm_op.cc 40 auto mean_flat = mean.flat<T1>(); local
59 QuantizedToFloat(mean_flat(channel), mean_min, mean_max);
103 auto mean_flat = mean.flat<T1>(); local
124 QuantizedToFloat(mean_flat(channel), mean_min, mean_max);
adjust_contrast_op.cc 239 TTypes<float, 1>::Tensor mean_flat(&mean(0, 0), mean.size());
310 auto mean_plane = mean_flat.slice(Index(i * channels), Index(channels));

Completed in 100 milliseconds