Home
last modified time | relevance | path

Searched defs:r_mean (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/evaluation/proto/
H A Dpreprocessing_steps.proto97 required float r_mean = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
H A Dimage_preprocessing_stage.h139 void AddPerChannelNormalizationStep(float r_mean, float g_mean, float b_mean, in AddPerChannelNormalizationStep()
H A Dimage_preprocessing_stage.cc227 float r_mean = params.means().r_mean(); in Normalize() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh485 stat_accscalar_t r_mean = mean[plane]; in batch_norm_backward_reduce_kernel() local
1207 auto r_mean = mean[c_offset]; in batch_norm_backward_reduce_channels_last_kernel() local