Searched refs:output_index_ (Results 1 – 1 of 1) sorted by relevance
411 : context_(context), output_index_(output_index) {} in CudnnRnnAllocatorInOutput()424 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()439 int output_index_; member in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput