Searched refs:CudnnBatchNormAllocatorInTemp (Results 1 – 1 of 1) sorted by relevance
664 class CudnnBatchNormAllocatorInTemp : public ScratchAllocator { class666 ~CudnnBatchNormAllocatorInTemp() override = default;668 explicit CudnnBatchNormAllocatorInTemp(OpKernelContext* context) in CudnnBatchNormAllocatorInTemp() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp924 std::unique_ptr<functor::CudnnBatchNormAllocatorInTemp<uint8>> in operator ()()930 new functor::CudnnBatchNormAllocatorInTemp<uint8>(context)); in operator ()()1106 std::unique_ptr<functor::CudnnBatchNormAllocatorInTemp<uint8>> in operator ()()1114 new functor::CudnnBatchNormAllocatorInTemp<uint8>(context)); in operator ()()