get_allocated_tensor (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A D	cudnn_rnn_ops.cc	`394 Tensor get_allocated_tensor(int index) const { in get_allocated_tensor() function in tensorflow::__anon980e2df00111::CudnnRnnAllocatorInTemp 1777 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutotune()`
H A D	fused_batch_norm_op.cc	`706 Tensor get_allocated_tensor(int index) const { in get_allocated_tensor() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp`