Home
last modified time | relevance | path

Searched defs:device_memory_size (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dautotune_maps_utils.cc48 uint64 device_memory_size = -1; in DeviceIdToIdentifierHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddevice_description.h251 int64_t device_memory_size() const { return device_memory_size_; } in device_memory_size() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op.cc86 size_t device_memory_size) { in AllocateDeviceMemory()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc932 uint64_t device_memory_size = -1; in CreateDeviceDescription() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc1069 uint64_t device_memory_size = -1; in CreateDeviceDescription() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h140 int64_t device_memory_size; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/
H A Draw_api_test.cc2113 int64_t device_memory_size = 8LL * 1024 * 1024 * 1024; in TEST() local