Searched defs:allocated_tensors_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | conv_ops_gpu.h | 95 std::vector<Tensor> allocated_tensors_; variable |
| D | fft_ops.cc | 396 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anonbbe6ef150111::CufftScratchAllocator |
| D | fused_batch_norm_op.cc | 703 std::vector<Tensor> allocated_tensors_; member in tensorflow::functor::CudnnBatchNormAllocatorInTemp |
| D | cudnn_rnn_ops.cc | 398 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp |