Home
last modified time | relevance | path

Searched defs:temporary_memory (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dtemporary_memory_manager.h137 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>> temporary_memory = in AllocateArray() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h49 Tensor temporary_memory; in AllocateBytes() local
Dfft_ops.cc321 Tensor temporary_memory; in AllocateBytes() local
Dcudnn_rnn_ops.cc372 Tensor temporary_memory; in AllocateBytes() local
422 Tensor* temporary_memory = nullptr; in AllocateBytes() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dcost_estimator.h143 int64 temporary_memory; member