Searched refs:temporary_memory (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | temporary_memory_manager.h | 137 port::StatusOr<std::unique_ptr<TemporaryDeviceMemoryBase>> temporary_memory = in AllocateArray() local
139 if (!temporary_memory.ok()) { in AllocateArray()
140 return temporary_memory.status(); in AllocateArray()
145 temporary_memory.ConsumeValueOrDie().release())); in AllocateArray()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 49 Tensor temporary_memory; in AllocateBytes() local
60 DT_UINT8, TensorShape({byte_size}), &temporary_memory, in AllocateBytes()
67 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
70 AsDeviceMemory(temporary_memory.flat<uint8>().data(), in AllocateBytes()
71 temporary_memory.flat<uint8>().size())); in AllocateBytes()
|
| D | fft_ops.cc | 321 Tensor temporary_memory; in AllocateBytes() local
328 DT_UINT8, TensorShape({byte_size}), &temporary_memory, in AllocateBytes()
335 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
338 AsDeviceMemory(temporary_memory.flat<uint8>().data(), in AllocateBytes()
339 temporary_memory.flat<uint8>().size())); in AllocateBytes()
|
| D | batch_matmul_op_impl.h | 267 Tensor temporary_memory;
270 DT_UINT8, TensorShape({byte_size}), &temporary_memory));
277 allocated_tensors_.push_back(temporary_memory);
280 temporary_memory.flat<uint8>().data(),
281 temporary_memory.flat<uint8>().size()));
|
| D | cudnn_rnn_ops.cc | 372 Tensor temporary_memory; in AllocateBytes() local
377 tf_data_type, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
383 allocated_tensors_.push_back(temporary_memory); in AllocateBytes()
386 temporary_memory.template flat<T>().data(), in AllocateBytes()
387 temporary_memory.template flat<T>().size() * sizeof(T)); in AllocateBytes()
422 Tensor* temporary_memory = nullptr; in AllocateBytes() local
424 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
430 temporary_memory->template flat<T>().data(), in AllocateBytes()
431 temporary_memory->template flat<T>().size() * sizeof(T)); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | cost_estimator.h | 143 int64 temporary_memory; member
186 temporary_memory = kMemoryUnknown; in Costs()
199 costs.temporary_memory = kZeroMemory; in ZeroCosts()
|