Searched defs:CudnnBatchNormAllocatorInOutput (Results 1 – 1 of 1) sorted by relevance
711 class CudnnBatchNormAllocatorInOutput : public ScratchAllocator { class721 CudnnBatchNormAllocatorInOutput(OpKernelContext* context, int output_index) in CudnnBatchNormAllocatorInOutput() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput