Home
last modified time | relevance | path

Searched refs:kCudnnBatchNormForwardTraining (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.cc49 case Thunk::kCudnnBatchNormForwardTraining: in ThunkKindToString()
Dthunk.h54 kCudnnBatchNormForwardTraining, enumerator
Dcudnn_batchnorm_thunk.cc80 : Thunk(Thunk::Kind::kCudnnBatchNormForwardTraining, thunk_info), in CudnnBatchNormForwardTrainingThunk()