Searched defs:allocated_tensors_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | conv_ops_gpu.h | 91 std::vector<Tensor> allocated_tensors_; variable |
| D | fft_ops.cc | 407 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anon4f5ed94b0111::CufftScratchAllocator |
| D | fused_batch_norm_op.cc | 713 std::vector<Tensor> allocated_tensors_; member in tensorflow::functor::CudnnBatchNormAllocatorInTemp |
| D | cudnn_rnn_ops.cc | 401 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anonf4c408700111::CudnnRnnAllocatorInTemp |