Searched defs:output_mean (Results 1 – 2 of 2) sorted by relevance
140 const BufferAllocation::Slice& output_mean, in CudnnBatchNormForwardTrainingThunk()176 se::DeviceMemory<float> output_mean( in ExecuteOnStream() local
418 auto output_mean = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); in HandleCustomCall() local