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