Searched refs:output_index_ (Results 1 – 2 of 2) sorted by relevance
716 OP_REQUIRES_OK(context_, context_->allocate_output(output_index_, {}, in ~CudnnBatchNormAllocatorInOutput()722 : context_(context), output_index_(output_index) {} in CudnnBatchNormAllocatorInOutput()737 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()753 int output_index_; member in tensorflow::functor::CudnnBatchNormAllocatorInOutput
410 : context_(context), output_index_(output_index) {} in CudnnRnnAllocatorInOutput()422 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()437 int output_index_; member in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput