Searched refs:CudnnBatchNormAllocatorInOutput (Results 1 – 1 of 1) sorted by relevance
711 class CudnnBatchNormAllocatorInOutput : public ScratchAllocator { class713 ~CudnnBatchNormAllocatorInOutput() override { in ~CudnnBatchNormAllocatorInOutput()721 CudnnBatchNormAllocatorInOutput(OpKernelContext* context, int output_index) in CudnnBatchNormAllocatorInOutput() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput922 std::unique_ptr<functor::CudnnBatchNormAllocatorInOutput<U>> in operator ()()928 new functor::CudnnBatchNormAllocatorInOutput<U>(context, 5)); in operator ()()