Home
last modified time | relevance | path

Searched refs:get_allocated_tensor (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc391 Tensor get_allocated_tensor(int index) const { in get_allocated_tensor() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp
1733 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutoTune()
Dfused_batch_norm_op.cc696 Tensor get_allocated_tensor(int index) const { in get_allocated_tensor() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp