Searched defs:output_mean (Results 1 – 3 of 3) sorted by relevance
78 const BufferAllocation::Slice& output_mean, in CudnnBatchNormForwardTrainingThunk()96 se::DeviceMemory<float> output_mean( in ExecuteOnStream() local
51 se::DeviceMemory<float> output_mean; member257 se::DeviceMemoryBase output_data, se::DeviceMemory<float> output_mean, in RunCudnnBatchNormForwardTraining()
145 auto output_mean = in CreateSimpleFusedBatchNorm() local816 auto* output_mean = context.graph_view->GetNode("output_mean"); in TEST_F() local