Searched defs:output_inv_stddev (Results 1 – 2 of 2) sorted by relevance
79 const BufferAllocation::Slice& output_inv_stddev) in CudnnBatchNormForwardTrainingThunk()98 se::DeviceMemory<float> output_inv_stddev( in ExecuteOnStream() local
52 se::DeviceMemory<float> output_inv_stddev; member258 se::DeviceMemory<float> output_inv_stddev, se::DeviceMemory<float> scale, in RunCudnnBatchNormForwardTraining()