Searched refs:CudnnBatchNormForwardTrainingParams (Results 1 – 1 of 1) sorted by relevance
47 struct CudnnBatchNormForwardTrainingParams { struct162 CudnnBatchNormForwardTrainingParams* params, se::Stream* stream) { in RunCudnnBatchNormForwardTrainingImpl()260 CudnnBatchNormForwardTrainingParams forward_params; in RunCudnnBatchNormForwardTraining()