/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
H A D | gpu_device_factory.cc | 32 Bytes memory_limit, const DeviceLocality& locality, in GPUDevice() 65 const SessionOptions& options, const string& name, Bytes memory_limit, in CreateGPUDevice() 84 Bytes memory_limit, const DeviceLocality& locality, in GPUCompatibleCPUDevice()
|
H A D | gpu_device.cc | 388 Bytes memory_limit, const DeviceLocality& locality, in BaseGPUDevice() 1087 int64_t* memory_limit) { in SingleVirtualDeviceMemoryLimit() 1561 TfDeviceId tf_device_id, int64_t memory_limit, in CreateGPUDevice()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
H A D | pluggable_device_factory.cc | 81 int64_t* memory_limit) { in SingleVirtualDeviceMemoryLimit() 227 TfDeviceId tf_device_id, int64_t memory_limit, in CreatePluggableDevice()
|
H A D | pluggable_device.cc | 142 Bytes memory_limit, const DeviceLocality& locality, TfDeviceId tf_device_id, in PluggableDevice()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | data_flow_ops.py | 1618 memory_limit=0): argument 1676 def memory_limit(self): member in BaseStagingArea 1860 memory_limit=0): argument 2113 memory_limit=0): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | stage_op.cc | 37 explicit Buffer(std::size_t capacity, std::size_t memory_limit) in Buffer() 186 int64_t memory_limit; in GetBuffer() local
|
H A D | map_stage_op.cc | 347 std::size_t memory_limit) in StagingMap() 504 int64_t memory_limit; in GetStagingMap() local
|
H A D | conv_ops_gpu.h | 50 DnnScratchAllocator(int64_t memory_limit, OpKernelContext* context) in DnnScratchAllocator()
|
H A D | fft_ops.cc | 376 CufftScratchAllocator(int64_t memory_limit, OpKernelContext* context) in CufftScratchAllocator()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | device.cc | 44 const string& name, DeviceType device, Bytes memory_limit, in BuildDeviceAttributes()
|
H A D | device_attributes.proto | 41 int64 memory_limit = 4; field
|
H A D | device.h | 170 const std::string& name, DeviceType device, Bytes memory_limit, in BuildDeviceAttributes()
|
/aosp_15_r20/external/libchrome/base/ |
H A D | sys_info.cc | 39 size_t memory_limit = kLowMemoryDeviceThresholdMB * 1024 * 1024; in AmountOfAvailablePhysicalMemory() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
H A D | stat_summarizer_options.h | 37 int memory_limit; member
|
H A D | test_log.proto | 133 int64 memory_limit = 3; // Memory capacity in bytes. field
|
/aosp_15_r20/external/iproute2/tc/ |
H A D | q_fq_codel.c | 165 unsigned int memory_limit; in fq_codel_print_opt() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/ |
H A D | profile.proto | 43 uint64 memory_limit = 3; field
|
/aosp_15_r20/external/perfetto/protos/third_party/pprof/ |
H A D | profile.proto | 156 uint64 memory_limit = 3; field
|
/aosp_15_r20/external/cronet/base/system/ |
H A D | sys_info.cc | 70 uint64_t memory_limit = kLowMemoryDeviceThresholdMB * 1024 * 1024; in AmountOfAvailablePhysicalMemory() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
H A D | tpu_platform.cc | 168 Status TpuPlatform::TpuMemoryLimit(int64_t* memory_limit) { in TpuMemoryLimit()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | threadpool_device.cc | 57 const string& name, Bytes memory_limit, in ThreadPoolDevice()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
H A D | redzone_allocator.cc | 53 int64_t memory_limit, int64_t redzone_size, in RedzoneAllocator()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
H A D | xla_tpu_device.cc | 442 int64_t memory_limit; in CreateDevices() local
|
/aosp_15_r20/external/perfetto/src/trace_processor/util/ |
H A D | profile_builder.h | 229 const uint64_t memory_limit; member
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/ |
H A D | benchmark_model.cc | 394 int memory_limit = 10; in Main() local
|