Home
last modified time | relevance | path

Searched refs:allocated_tensors_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h67 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
79 std::vector<Tensor> allocated_tensors_; variable
Dfft_ops.cc335 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
347 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anonbbe6ef150111::CufftScratchAllocator
Dbatch_matmul_op_impl.h277 allocated_tensors_.push_back(temporary_memory);
286 std::vector<Tensor> allocated_tensors_;
Dcudnn_rnn_ops.cc383 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
393 return allocated_tensors_[index]; in get_allocated_tensor()
399 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp