Searched defs:CudnnBatchNormAllocatorInTemp (Results 1 – 1 of 1) sorted by relevance
664 class CudnnBatchNormAllocatorInTemp : public ScratchAllocator { class668 explicit CudnnBatchNormAllocatorInTemp(OpKernelContext* context) in CudnnBatchNormAllocatorInTemp() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp