Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h83 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
95 std::vector<Tensor> allocated_tensors_; variable
Dfft_ops.cc384 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
396 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anon049b3f3c0111::CufftScratchAllocator
Dmatmul_op_impl.h309 allocated_tensors_.push_back(temporary_memory);
318 std::vector<Tensor> allocated_tensors_;
Dfused_batch_norm_op.cc687 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
697 return allocated_tensors_[index]; in get_allocated_tensor()
703 std::vector<Tensor> allocated_tensors_; member in tensorflow::functor::CudnnBatchNormAllocatorInTemp
Dcudnn_rnn_ops.cc382 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
392 return allocated_tensors_[index]; in get_allocated_tensor()
398 std::vector<Tensor> allocated_tensors_; member in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInTemp