OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:DeviceMemoryUsage
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/core/grappler/
devices.cc
64
CHECK(se->
DeviceMemoryUsage
(&available_memory, &total_memory));
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
executor.h
164
bool
DeviceMemoryUsage
(int64 *free, int64 *total) const override {
/external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.h
167
bool
DeviceMemoryUsage
(int64 *free, int64 *total) const override;
cuda_gpu_executor.cc
783
bool CUDAExecutor::
DeviceMemoryUsage
(int64 *free, int64 *total) const {
[
all
...]
/external/tensorflow/tensorflow/stream_executor/host/
host_gpu_executor.h
146
bool
DeviceMemoryUsage
(int64 *free, int64 *total) const override {
/external/tensorflow/tensorflow/stream_executor/
stream_executor_internal.h
232
virtual bool
DeviceMemoryUsage
(int64 *free, int64 *total) const {
stream_executor_pimpl.cc
706
bool StreamExecutor::
DeviceMemoryUsage
(int64 *free, int64 *total) const {
707
return implementation_->
DeviceMemoryUsage
(free, total);
stream_executor_pimpl.h
330
bool
DeviceMemoryUsage
(int64 *free, int64 *total) const;
[
all
...]
/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc
789
if (!se->
DeviceMemoryUsage
(&available_memory, &total_memory)) {
[
all
...]
Completed in 314 milliseconds