Searched refs:kCudnnBatchNormForwardTraining (Results 1 – 3 of 3) sorted by relevance
35 case Thunk::kCudnnBatchNormForwardTraining: in ThunkKindToString()
51 kCudnnBatchNormForwardTraining, enumerator
143 : Thunk(Thunk::Kind::kCudnnBatchNormForwardTraining, hlo), in CudnnBatchNormForwardTrainingThunk()